input: 'cg': nodes=species=Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, cart=Array([[-1.6074809 , 3.990498 , 0.59335387],
[13.654599 , 0.6031299 , 4.7416563 ],
[ 2.6521325 , 3.3873684 , 4.7416563 ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, graph_i=Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
,), edges=to_jimage=Array([[[-1, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, -1]],
[[ 0, -1, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 1, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
...,
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]]], dtype=int8):Rendering array...
, receiver=Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,), n_node=Array([168, 80, 68, 60, 53, 48, 42, 40, 36, 32, 31, 28, 26,
24, 24, 21, 20, 18, 16, 16, 14, 12, 12, 10, 9, 8,
6, 5, 4, 4, 1, 88], dtype=uint16):Rendering array...
, padding_mask=Array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, False], dtype=bool):Rendering array...
, graph_data=dataset_id=Array([ 114536034, 1780206004, -1687275888, 1202003, 1758626004,
1556291002, 139460000, 1554815003, 118640006, 1675109005,
-1651451869, 1555003018, 1615850000, -1709938884, -1592934888,
-1650297867, -1696810885, -1663817880, -1696683861, -1710558886,
-1662911874, -1781222858, 123675020, -1806109888, -1804816886,
-1875814888, -1668181878, 122133000, 1864969002, -1700082886,
120483000, 0], dtype=int32):Rendering array...
, abc=Array([[17.914213 , 17.914213 , 8.296605 ],
[10.275203 , 16.84246 , 5.0821056],
[ 6.74365 , 9.197391 , 12.196357 ],
[ 9.433409 , 9.433409 , 20.947056 ],
[10.816606 , 10.426013 , 10.417979 ],
[ 6.8768296, 11.247085 , 19.015215 ],
[ 9.827564 , 9.827564 , 6.762959 ],
[ 6.386287 , 6.386287 , 12.2509365],
[ 8.614349 , 8.614349 , 8.614349 ],
[ 7.9092207, 7.7653685, 6.9275327],
[ 8.073368 , 20.393888 , 10.60539 ],
[ 5.306318 , 9.112615 , 10.040712 ],
[ 5.627906 , 6.953792 , 10.501416 ],
[ 5.037883 , 5.8846297, 7.8679 ],
[ 5.66 , 5.650074 , 9.156885 ],
[ 6.1716037, 8.621225 , 5.767166 ],
[ 6.2029986, 8.613454 , 8.613454 ],
[ 5.494526 , 5.494526 , 7.4353304],
[ 3.7546806, 4.811737 , 5.4126735],
[ 2.715433 , 5.2583437, 9.388806 ],
[ 6.0365143, 6.034208 , 6.02813 ],
[ 5.419802 , 6.49084 , 6.660221 ],
[ 4.117953 , 5.849566 , 5.849566 ],
[ 4.299559 , 4.299559 , 10.174184 ],
[ 4.235417 , 7.5434947, 7.5435066],
[ 6.565039 , 6.565039 , 6.565039 ],
[ 4.3139734, 5.5680494, 5.484443 ],
[ 5.9383936, 5.9383936, 5.9383936],
[ 4.4320164, 4.4320164, 4.4320164],
[ 3.640625 , 3.640625 , 6.0473022],
[ 3.571267 , 3.571267 , 3.571267 ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, angles_rad=Array([[1.5707964, 1.5707964, 2.0943952],
[1.5707964, 1.5707964, 1.5707964],
[1.2343053, 1.5818529, 1.1629694],
[1.5707964, 1.5707964, 1.5707964],
[1.8930188, 1.8607063, 2.0151982],
[1.7593405, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 2.0943952],
[1.730846 , 1.730846 , 1.8141689],
[1.0478324, 1.0478324, 1.0478324],
[1.8424883, 1.5707964, 1.5707964],
[0.3288564, 1.0035257, 1.0851887],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[1.7750735, 1.7689804, 1.2668823],
[1.6448978, 1.4945464, 1.8871104],
[1.5798758, 1.5553586, 1.5304748],
[1.2806493, 1.2024467, 1.2024467],
[1.2456238, 1.397332 , 1.9114691],
[1.5707964, 1.5707964, 1.5707964],
[1.644386 , 1.5707964, 1.5707964],
[1.5746922, 2.094859 , 1.0429444],
[2.0565953, 1.3555512, 1.5138267],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[2.0943942, 1.5707964, 1.5707964],
[2.2965217, 2.2965217, 1.237292 ],
[1.6661328, 1.9750055, 1.1730014],
[2.4071865, 2.4071865, 1.0650531],
[1.0471976, 1.0471976, 1.0471976],
[1.5707964, 1.5707964, 2.0943952],
[1.0471976, 1.0471976, 1.0471976],
[0. , 0. , 0. ]], dtype=float32):Rendering array...
, lat=Array([[[ 1.7914213e+01, 1.5000001e-07, 0.0000000e+00],
[-8.9571066e+00, 1.5514163e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 8.2966051e+00]],
[[ 1.0275203e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.6842461e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.0821056e+00]],
[[ 1.6343699e-01, -1.7300400e-01, 6.7394490e+00],
[ 8.5276871e+00, 6.0145002e-02, 3.4448421e+00],
[ 4.2400112e+00, 1.1435485e+01, 5.5798002e-02]],
[[ 9.4334087e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.4334087e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 2.0947056e+01]],
[[-6.3495708e+00, 6.5125003e+00, 5.8539939e+00],
[ 6.0938725e+00, -6.4372239e+00, 5.4889545e+00],
[ 6.0836735e+00, 6.1772451e+00, -5.7762318e+00]],
[[ 6.8768296e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.1048177e+01, -2.1058772e+00],
[ 0.0000000e+00, -3.7062699e-03, 1.9015215e+01]],
[[ 4.9137821e+00, -8.5109196e+00, 0.0000000e+00],
[ 4.9137821e+00, 8.5109196e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7629590e+00]],
[[ 6.3021402e+00, -6.3853199e-03, -1.0332731e+00],
[-1.7227671e+00, 6.0621023e+00, -1.0332731e+00],
[ 3.0169221e-02, 3.9895471e-02, 1.2250834e+01]],
[[ 7.4619780e+00, -1.4151999e-03, 4.3041716e+00],
[ 2.4855030e+00, 7.0358648e+00, 4.3041716e+00],
[-2.0012499e-03, -1.4151999e-03, 8.6143484e+00]],
[[ 7.9092207e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 7.5781007e+00, -1.6950918e+00],
[ 0.0000000e+00, -3.5751951e-01, 6.9183011e+00]],
[[ 6.7416244e+00, 5.3566109e-02, 4.4414983e+00],
[-1.2468886e+00, 7.0104933e+00, 1.9110441e+01],
[-1.6173536e-01, 2.7245510e-01, 1.0600656e+01]],
[[ 5.3063178e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.1126146e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0040712e+01]],
[[ 5.6279058e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 6.9537921e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0501416e+01]],
[[ 4.9857669e+00, 3.5377501e-03, -7.2275919e-01],
[ 1.6229881e+00, 5.5577154e+00, -1.0519388e+00],
[-4.2640892e-01, -7.8503363e-02, 7.8559442e+00]],
[[-4.6255998e-02, 4.7460151e+00, 3.0836339e+00],
[ 4.7340288e+00, -4.6193000e-02, -3.0838571e+00],
[-4.7798352e+00, 4.7923450e+00, -6.1672649e+00]],
[[ 6.1701546e+00, 1.2492373e-01, 4.7747571e-02],
[ 1.7339531e-01, 8.6193953e+00, -3.8590480e-02],
[ 4.4977799e-02, -2.7458400e-02, 5.7669253e+00]],
[[ 0.0000000e+00, 9.9999999e-09, -6.2029986e+00],
[-5.1461577e+00, 6.1716561e+00, -3.1015000e+00],
[ 5.1461577e+00, 6.1716561e+00, -3.1015000e+00]],
[[-3.9119451e+00, -3.1703908e+00, 2.1988914e+00],
[ 3.9119451e+00, -3.1703908e+00, -2.1988914e+00],
[ 3.8619483e+00, -3.1703908e+00, 5.5061889e+00]],
[[ 0.0000000e+00, -3.7546806e+00, 0.0000000e+00],
[-4.8117371e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -5.4126735e+00]],
[[ 2.7154329e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.2441120e+00, -3.8661101e-01],
[ 0.0000000e+00, 0.0000000e+00, 9.3888063e+00]],
[[-4.2616282e+00, -8.0763102e-03, -4.2752743e+00],
[-8.2218703e-03, -4.2585020e+00, -4.2751322e+00],
[-3.6448999e-03, -4.2542191e+00, 4.2708263e+00]],
[[ 5.3959742e+00, 2.2922444e-01, -4.5296320e-01],
[-7.0430361e-02, -1.8796436e+00, -6.2123251e+00],
[ 1.2653049e+00, 6.4092894e+00, 1.2955921e+00]],
[[ 4.1179528e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.8495660e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.8495660e+00]],
[[ 4.2995591e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995591e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0174184e+01]],
[[ 0.0000000e+00, 0.0000000e+00, -4.2354169e+00],
[-3.7717474e+00, -6.5328579e+00, 0.0000000e+00],
[-3.7717595e+00, 6.5328650e+00, 0.0000000e+00]],
[[-2.6921539e+00, 2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, -2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, 2.6921539e+00, -5.3483038e+00]],
[[-4.3139734e+00, 0.0000000e+00, 0.0000000e+00],
[-2.1569867e+00, -3.5078707e+00, 3.7477229e+00],
[ 2.1569867e+00, 3.4406567e+00, 3.6862457e+00]],
[[-2.1319211e+00, 2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, -2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, 2.1319211e+00, -5.1160870e+00]],
[[ 0.0000000e+00, 3.1339090e+00, 3.1339090e+00],
[ 3.1339090e+00, 0.0000000e+00, 3.1339090e+00],
[ 3.1339090e+00, 3.1339090e+00, 0.0000000e+00]],
[[ 1.8203124e+00, -3.1528738e+00, 0.0000000e+00],
[ 1.8203124e+00, 3.1528738e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.0473022e+00]],
[[ 0.0000000e+00, 2.5252669e+00, 2.5252669e+00],
[ 2.5252669e+00, 0.0000000e+00, 2.5252669e+00],
[ 2.5252669e+00, 2.5252669e+00, 0.0000000e+00]],
[[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00]]], dtype=float32):Rendering array...
,), target_data=e_form=Array([-6.45586 , -6.8128195, -6.207207 , -5.4671555, -6.7837076,
-5.1435127, -7.477968 , -8.923818 , -8.200472 , -7.9467635,
-9.577978 , -6.3009315, -6.341573 , -6.54503 , -9.78936 ,
-8.184659 , -6.8508086, -5.8719406, -4.560977 , -6.476559 ,
-8.410168 , -7.7028112, -4.6160893, -7.076347 , -4.2489 ,
-4.8315697, -4.6079087, -5.067987 , -4.573908 , -4.5372405,
-3.712647 , 0. ], dtype=float32):Rendering array...
, force=Array([[ 0.01931927, -0.00101004, 0.00514533],
[ 0.01053436, 0.01622596, 0.00514533],
[ 0.00878491, -0.017236 , 0.00514533],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, stress=Array([[[-5.7650656e-01, 1.3470000e-05, 0.0000000e+00],
[-1.3470000e-05, -5.7650650e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8034226e-01]],
[[ 1.2198363e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0798175e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 7.2570233e+00]],
[[-6.6261089e-01, -1.3693068e+00, 1.7417374e-01],
[-1.3692706e+00, 9.3778890e-01, 5.2438201e-03],
[ 1.7419136e-01, 5.2425899e-03, -9.8206145e-01]],
[[-2.8648193e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -2.8648193e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8271584e+01]],
[[-3.5376236e+00, -6.9719963e-02, -1.5628552e-01],
[-6.9719352e-02, -3.4879370e+00, -6.0669428e-01],
[-1.5628555e-01, -6.0669398e-01, -2.6510825e+00]],
[[ 4.2228210e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 3.6939880e-01, 2.1828409e-01],
[ 0.0000000e+00, 2.1827926e-01, 7.3054400e-03]],
[[ 5.8439370e-02, 2.6000001e-07, -0.0000000e+00],
[-2.6000001e-07, 5.8439370e-02, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.5984979e+00]],
[[-5.6706039e+01, 2.8917530e+00, -1.5447336e+01],
[ 2.8917530e+00, -5.5068783e+01, -2.0427404e+01],
[-1.5447323e+01, -2.0427385e+01, -2.5514915e+01]],
[[-5.2726287e-01, -5.5939171e-02, -1.3698542e-01],
[-5.5939171e-02, -4.8771665e-01, -9.6870460e-02],
[-1.3698542e-01, -9.6870460e-02, -6.8537790e-01]],
[[ 5.0960288e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.7661177e+01, 8.4236950e-01],
[ 0.0000000e+00, 8.4236366e-01, -8.4243935e-01]],
[[-3.1354368e+00, -2.1732990e-02, -1.6962990e-02],
[-2.1732951e-02, -3.8033817e+00, 3.7452817e+00],
[-1.6962999e-02, 3.7452817e+00, -1.9895161e+00]],
[[ 1.7607585e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0304039e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.6413042e-01]],
[[-2.0985847e+00, -1.6143000e-04, -4.7509999e-05],
[-1.6518999e-04, -1.3785554e+00, -1.3842000e-04],
[-5.1700001e-05, -1.3332001e-04, 8.3491117e-01]],
[[-9.3856964e+00, -3.4404862e-01, -2.0616198e+00],
[-3.4404868e-01, -7.5551205e+00, -7.3723370e-01],
[-2.0616195e+00, -7.3723364e-01, -1.1444527e+01]],
[[-1.9861711e+00, -2.8908322e+00, 7.8711361e-01],
[-2.8947139e+00, -4.8037186e+00, 2.2988582e-01],
[ 7.8879166e-01, 2.2843994e-01, 2.1315180e+01]],
[[-1.4272230e+00, -1.6913697e+00, -5.8565631e+00],
[-1.6913695e+00, -4.9773030e+00, -1.3547415e+00],
[-5.8565636e+00, -1.3547415e+00, 1.7153270e+00]],
[[-6.0113930e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -8.9604294e+01, 7.7699997e-06],
[ 0.0000000e+00, 7.7699997e-06, -6.6187057e+01]],
[[ 2.7452025e+00, -0.0000000e+00, 4.3098168e+00],
[-0.0000000e+00, 9.5339581e-02, 0.0000000e+00],
[ 4.3098168e+00, 0.0000000e+00, -4.1897740e+00]],
[[-1.9521265e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -6.4751107e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.5113559e-01]],
[[ 2.3531018e+02, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.1068205e+01, -1.0828968e+02],
[ 0.0000000e+00, -1.0828968e+02, 1.9093062e+02]],
[[-4.9796581e+00, -1.2185392e+00, -4.9542028e-01],
[-1.2196398e+00, -2.3808525e+00, -6.0947609e-01],
[-4.9716067e-01, -6.1348146e-01, -7.8811878e-01]],
[[ 8.0838509e+00, 1.0610949e+01, -6.8462425e-01],
[ 1.0610948e+01, 1.6280436e+01, -5.4885068e+00],
[-6.8462473e-01, -5.4885073e+00, -1.0486948e+01]],
[[ 1.4801001e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6347537e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.6347537e+00]],
[[ 4.2995720e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995720e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7877603e+00]],
[[-5.6527023e+00, -1.9787040e-02, 0.0000000e+00],
[-1.9787040e-02, -5.6755505e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.4912200e+00]],
[[-1.5253462e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.5253462e+00, -0.0000000e+00],
[-0.0000000e+00, -0.0000000e+00, 3.4477546e+00]],
[[-1.2089447e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6878867e+00, -2.0085368e+00],
[ 0.0000000e+00, -2.0085406e+00, 3.9861345e+00]],
[[ 1.3792354e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.3792354e-01, -0.0000000e+00],
[ 0.0000000e+00, -0.0000000e+00, 2.5862169e-01]],
[[ 1.4818479e+01, 1.4620157e+00, 1.4620157e+00],
[ 1.4620157e+00, 1.4818479e+01, 1.4620157e+00],
[ 1.4620157e+00, 1.4620157e+00, 1.4818479e+01]],
[[-3.0764341e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -3.0764344e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, 4.8425870e+00]],
[[-1.4238837e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.4238837e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, -1.4238837e+00]],
[[ 1.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0000000e+00]]], dtype=float32):Rendering array...
,),),}children: 'mace': maceinput: 'vectors': Array([[[-0.08514988, -1.5187607 , 0.35024124],
[-1.322167 , 1.2215691 , 0.7999469 ],
[ 1.3537681 , -2.1077032 , 0.3074897 ],
...,
[ 3.0676713 , 3.6979523 , 0.7337971 ],
[ 1.6074809 , -3.990498 , 2.275238 ],
[ 0.06579673, 4.6275563 , -1.5481044 ]],
[[ 1.2727108 , -0.83312184, 0.35024118],
[-1.7189932 , -0.53424555, 0.7999468 ],
[ 2.5022097 , 0.11854595, 0.30748987],
...,
[-1.668685 , 4.505657 , 0.7337971 ],
[ 4.259614 , -0.60312974, 2.275238 ],
[-3.9746828 , 2.37076 , -1.548104 ]],
[[-1.3578603 , -0.68563867, 0.35024118],
[ 0.39682627, 1.7558143 , 0.7999468 ],
[-1.1484407 , -2.2262492 , 0.30748987],
...,
[ 4.736356 , -0.8077049 , 0.7337971 ],
[-2.6521325 , -3.3873684 , 2.275238 ],
[ 4.04048 , 2.2567966 , -1.548104 ]],
...,
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]]], dtype=float32):Rendering array...
, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,}children: 'node_embedding': node_embeddinginput: 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
,}children: 'embed': embedinput: 'inputs': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
,}params: 11520 parametersembedding: Array([[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-1.9298473e-01, 5.5534367e-02, 6.1264433e-02, ...,
7.9367173e-01, -6.8340577e-02, -2.3404005e-01],
[ 1.5960895e-01, -8.8038474e-02, 8.5871279e-02, ...,
9.9117720e-01, 1.6090073e-02, 1.2284823e-01],
...,
[ 4.0848142e-01, 4.4648898e-01, 1.2513540e-02, ...,
-3.9285350e-01, 8.4816426e-02, 1.3216788e-01],
[-7.4507594e-02, -1.0931188e+00, 8.0863041e-01, ...,
4.5552161e-01, 5.7509476e-01, 9.8894668e-01],
[-1.5040071e-01, -2.0703952e-01, 8.7037069e-01, ...,
-3.4091997e-01, -2.9804537e-01, -2.9061580e-01]], dtype=float32):Rendering array...
,}output: Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,),}output: 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,), 'edge_embedding': edge_embeddinginput: 'edge_lengths': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
,}children: 'basis': basisinput: 'x': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
, 'r_max': <jax.Array([5.], dtype=float32)>:Rendering array...
,}params: 8 parametersfreq: Array([0.37965846, 1.0474222 , 1.6848834 , 2.3603554 , 3.0224771 ,
3.7064195 , 4.424161 , 5.1440697 ], dtype=float32):Rendering array...
,}output: Array([[[ 0.14312896, 0.25572735, 0.12545101, ..., -0.12232928,
0.14945401, 0.25231254],
[ 0.13809775, 0.17893293, -0.03582633, ..., 0.17280407,
0.13234529, -0.13208821],
[ 0.12966888, 0.07571861, -0.14201179, ..., 0.01115712,
-0.1562002 , 0.06409264],
...,
[ 0.0792156 , -0.07647058, 0.07771403, ..., -0.06099052,
0.03856843, -0.01069126],
[ 0.07906061, -0.07617152, 0.07732467, ..., -0.05961608,
0.0364945 , -0.0080494 ],
[ 0.0787289 , -0.07552354, 0.07646646, ..., -0.05658155,
0.03198421, -0.00239656]],
[[ 0.14312896, 0.25572735, 0.12545101, ..., -0.12232928,
0.14945401, 0.25231254],
[ 0.13809775, 0.178933 , -0.03582628, ..., 0.17280397,
0.13234545, -0.13208795],
[ 0.12966886, 0.07571841, -0.14201187, ..., 0.01115651,
-0.15619996, 0.06409344],
...,
[ 0.0792156 , -0.07647058, 0.07771403, ..., -0.06099052,
0.03856843, -0.01069126],
[ 0.0790606 , -0.07617152, 0.07732463, ..., -0.05961608,
0.03649421, -0.00804925],
[ 0.0787289 , -0.07552354, 0.07646649, ..., -0.05658168,
0.0319845 , -0.00239672]],
[[ 0.14312896, 0.25572735, 0.12545094, ..., -0.12232928,
0.14945419, 0.25231248],
[ 0.13809773, 0.178933 , -0.03582619, ..., 0.17280397,
0.1323456 , -0.13208795],
[ 0.12966889, 0.07571857, -0.14201179, ..., 0.01115712,
-0.15620011, 0.0640929 ],
...,
[ 0.07921562, -0.0764706 , 0.07771403, ..., -0.06099075,
0.03856857, -0.01069141],
[ 0.07906061, -0.07617152, 0.07732467, ..., -0.05961608,
0.0364945 , -0.0080494 ],
[ 0.0787289 , -0.07552354, 0.07646646, ..., -0.05658155,
0.03198421, -0.00239656]],
...,
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]],
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]],
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]]], dtype=float32):Rendering array...
,), 'envelope': envelopeinput: 'r': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
, 'r_max': <jax.Array([5.], dtype=float32)>:Rendering array...
,}output: Array([[1. , 1. , 1. , ..., 0.20315367, 0.18754505,
0.15560311],
[1. , 1. , 1. , ..., 0.20315367, 0.18754408,
0.1556041 ],
[1. , 1. , 1. , ..., 0.20315467, 0.18754505,
0.15560311],
...,
[1. , 1. , 1. , ..., 1. , 1. ,
1. ],
[1. , 1. , 1. , ..., 1. , 1. ,
1. ],
[1. , 1. , 1. , ..., 1. , 1. ,
1. ]], dtype=float32):Rendering array...
,),}output: 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,), 'edge_embedding.avg_num_neighbors': edge_embeddinginput: 'metadata': dataset_name='mptrj', supported_targets='0': 'energy',}, batches_per_group=array([ 906, 1425, 1409, 1425, 1438, 1440, 1436, 1440, 1422, 1430, 1445,
1492, 1398, 1439, 1425, 1443, 1465, 1413, 1397, 1394, 1425, 1475,
1492, 1430, 1388, 1445, 1411, 1460, 1491, 1433, 1451, 1420, 1464,
1427, 1402, 1430], dtype=uint32):Rendering array...
, batch_num_atoms=32, nearest_k=48, batch_num_graphs=32, shift_energy=-6.198672771453857, scale_energy=2.3766870498657227, atomic_numbers=array([ 0, 6, 17, 3, 8, 15, 51, 12, 26, 29, 21, 31, 37, 83, 25, 23, 24,
16, 62, 32, 46, 34, 70, 19, 60, 9, 28, 7, 14, 20, 30, 52, 42, 27,
56, 38, 72, 41, 39, 47, 48, 50, 90, 40, 92, 58, 45, 55, 13, 53, 33,
82, 5, 22, 49, 11, 35, 78, 69, 73, 57, 93, 1, 94, 66, 77, 81, 4,
74, 64, 63, 75, 43, 59, 79, 44, 68, 80, 65, 67, 76, 36, 91, 71, 61,
54, 89, 10, 2, 18], dtype=uint8):Rendering array...
, atomwise_shift_energy=array([ 0. , -2.2707615 , 2.3401892 , 3.0822957 ,
-2.139031 , -0.16549587, 2.1411645 , 4.8762283 ,
-1.9809715 , 2.5143476 , -1.9116923 , 2.9952776 ,
4.3576717 , 2.7872267 , -2.9721358 , -3.6167579 ,
-3.5604646 , 0.6107417 , -1.0343803 , 1.5463392 ,
0.69618833, 1.4034783 , 2.7666323 , 4.2159247 ,
-0.9729802 , 0.4112059 , 0.38132888, -2.0058405 ,
0.34989974, 2.1912303 , 5.0034466 , 2.3049965 ,
-4.728218 , -0.54687583, 1.999384 , 2.2432775 ,
-5.1602435 , -4.790666 , -2.4661624 , 4.0280995 ,
5.34998 , 2.2807906 , -3.1929913 , -3.7777352 ,
-6.3766856 , -1.6896563 , -1.404504 , 4.1458993 ,
2.040268 , 3.405083 , 1.543168 , 2.8831744 ,
-1.0880722 , -3.9293804 , 3.4315777 , 4.092721 ,
2.8371766 , -0.34215775, -0.4136282 , -6.77352 ,
-1.7962555 , -6.870695 , 2.6937704 , -8.061367 ,
-0.5946495 , -2.793373 , 4.1255984 , 2.1051993 ,
-6.384699 , -10.322368 , -6.0671163 , -5.471272 ,
-1.4151858 , -0.7534348 , 2.9328883 , -2.8679254 ,
-0.39531928, 6.6045566 , -0.5138772 , -0.4270394 ,
-4.0246854 , 6.1608987 , -4.3657956 , -0.40262896,
0.736037 , 16.102348 , 1.9727042 , 0.7799125 ,
3.719906 , 1.1950303 ], dtype=float32):Rendering array...
, atomwise_scale_energy=array([-2.40991106e-39, 1.33058935e-01, 2.93497473e-01, 2.61184499e-02,
2.40968168e-01, 9.34929773e-02, 3.21534164e-02, 1.22204259e-01,
-3.99079267e-03, 3.21241319e-02, 5.05398333e-01, 1.65204424e-02,
1.08429275e-01, 1.10232485e-02, -2.18620170e-02, -2.22697780e-02,
-2.75362208e-02, 1.05140291e-01, 2.78376997e-01, 2.12133881e-02,
6.50004968e-02, 7.48609379e-02, 1.90245599e-01, 1.06534272e-01,
3.27279478e-01, 2.27959231e-01, 2.23455299e-02, 4.60880756e-01,
1.62185147e-01, 1.83875978e-01, 2.92971972e-02, 4.69289832e-02,
1.18544335e-02, 3.74471620e-02, 1.70294091e-01, 1.80884063e-01,
4.55600560e-01, 1.10464349e-01, 3.83205593e-01, 8.41733068e-02,
-6.63747604e-04, 1.12307980e-03, 3.71447802e-01, 3.02267253e-01,
1.66967139e-01, 2.51082033e-01, 1.04348332e-01, 8.47445428e-02,
1.59817740e-01, 1.23538606e-01, 7.84063041e-02, 8.03675130e-03,
1.46966413e-01, 2.89947659e-01, -1.36596169e-02, 7.84631744e-02,
1.75462380e-01, 5.37232794e-02, 2.80312657e-01, 2.79683918e-01,
3.19721878e-01, 2.42589667e-01, -3.06586875e-03, 2.37827122e-01,
2.86706179e-01, 7.04936832e-02, 3.83133627e-02, 6.03754342e-01,
-9.21594165e-03, 2.57200658e-01, 1.97187737e-01, 3.15266103e-01,
1.49181616e+00, 3.04980606e-01, 4.43397984e-02, 5.12347743e-02,
2.87493646e-01, 1.02658063e-01, 2.45865300e-01, 2.92902708e-01,
2.93616891e-01, 1.51683092e+00, 3.58057797e-01, 3.03377599e-01,
3.02371353e-01, 1.40909803e+00, 2.45032728e-01, 1.02921605e+00,
-7.56268263e-01, 1.25650644e+00], dtype=float32):Rendering array...
, r_max_quantile_r=array([ 0.5, 1. , 1.5, 2. , 2.5, 3. , 3.5, 4. , 4.5, 5. , 5.5,
6. , 6.5, 7. , 7.5, 8. , 8.5, 9. , 9.5, 10. , 10.5, 11. ,
11.5, 12. ], dtype=float32):Rendering array...
, r_max_quantile_k=array([[9.9999923e-01, 7.3028741e-07, 0.0000000e+00, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
[9.6614134e-01, 2.8218286e-02, 5.4964880e-03, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
[8.0479968e-01, 1.3872136e-01, 2.9714543e-02, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
...,
[1.4321747e-05, 4.8482966e-06, 1.3611745e-05, ..., 5.5298984e-05,
9.6763079e-06, 9.9777752e-01],
[9.0880203e-06, 3.8137230e-06, 1.0284881e-05, ..., 9.9745084e-05,
1.4930320e-05, 9.9792618e-01],
[6.5523009e-06, 2.3937198e-06, 7.8100175e-06, ..., 1.5727550e-04,
2.5276058e-05, 9.9818617e-01]], dtype=float32):Rendering array...
,),}output: <jax.Array([35.006073], dtype=float32)>:Rendering array...
,), 'layer_0': layer_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,}params: 28672 parametersw[0,0] 128x0e,224x0e: Array([[-0.15340874, 0.63586867, -2.066119 , ..., -1.2421888 ,
-0.564434 , -1.1749403 ],
[-0.4223909 , 1.9502263 , -1.2073642 , ..., 0.26299307,
1.3463928 , -0.4028095 ],
[ 1.0757334 , 0.45307207, 0.2272922 , ..., 1.7615929 ,
-1.8404636 , 0.5965187 ],
...,
[-1.5144392 , 1.1170446 , -1.6822397 , ..., -1.7941395 ,
-1.1314518 , -0.31909847],
[-0.03311044, 0.73373675, -0.40699175, ..., 0.63367414,
0.6495996 , -0.9303382 ],
[ 0.27654043, 0.6394971 , -0.8532704 , ..., -0.43343392,
-1.7429564 , 2.0256884 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,}params: 16384 parametersw[0,0] 128x0e,128x0e: Array([[-0.61885774, -1.0856944 , -1.6025343 , ..., -0.55894697,
-1.3429723 , 0.34905353],
[ 0.09641334, 0.12247575, 1.810319 , ..., 0.14283091,
1.1057113 , 0.82308096],
[-1.4454341 , -2.1964498 , 0.57032555, ..., 0.59593636,
2.0616703 , 1.0615976 ],
...,
[ 1.7831123 , -0.78575855, 0.907529 , ..., -2.4380598 ,
0.76305825, -1.559743 ],
[ 0.5572579 , -0.3384806 , -0.7951938 , ..., -0.7975023 ,
1.2298973 , -1.0195863 ],
[-0.37849867, -0.37187925, 2.0620265 , ..., 4.0443935 ,
1.5165552 , -0.24680647]], dtype=float32):Rendering array...
,}output: 128x0e
[[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
...
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]]key=0,): Array([[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
...,
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45]], dtype=float32), :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
...
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]]key=0,): Array([[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
...,
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 3072 parameterskernel: Array([[-1.6549976 , 1.8082368 , -0.931092 , ..., -1.1570392 ,
0.29403457, 2.0080574 ],
[-3.9210534 , 4.3420377 , -1.6388535 , ..., 0.42035243,
1.5823126 , 1.8510852 ],
[-5.891932 , 1.6466467 , -0.91515696, ..., 3.1036937 ,
0.06468228, -0.01008435],
...,
[-0.3021125 , 0.21281758, 2.282933 , ..., 2.4923284 ,
2.219395 , 0.38834897],
[ 1.2368158 , -0.5212946 , 1.5847732 , ..., 1.4557836 ,
0.837641 , 0.28724718],
[ 0.61822885, -0.6755894 , 0.49171826, ..., 0.6699977 ,
-0.3290476 , 0.2587202 ]], dtype=float32):Rendering array...
,}output: Array([[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177255e-01, 1.03016412e+00, -1.49667375e-02, ...,
-8.23826373e-01, 3.75961483e-01, 4.20294553e-01],
[-3.30404937e-02, 4.40512866e-01, -1.09523706e-01, ...,
-4.86856103e-01, 3.95714074e-01, 5.15792191e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669414e-02, ...,
-8.23826849e-01, 3.75960767e-01, 4.20294493e-01],
[-3.30404937e-02, 4.40511018e-01, -1.09523691e-01, ...,
-4.86855090e-01, 3.95713001e-01, 5.15792251e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40497576e-03, -8.86400975e-03, -6.06032414e-03, ...,
-2.22131051e-02, -1.79946907e-02, 3.61362263e-03],
[-7.22511625e-03, -7.52228592e-03, -4.98296879e-03, ...,
-1.78871062e-02, -1.52160879e-02, 3.11822048e-03]],
[[-7.04736188e-02, 1.21852052e+00, -9.89617407e-01, ...,
-9.95731711e-01, -7.08278000e-01, 3.97574872e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669452e-02, ...,
-8.23826671e-01, 3.75960618e-01, 4.20294344e-01],
[-3.30405124e-02, 4.40512449e-01, -1.09523349e-01, ...,
-4.86855447e-01, 3.95714134e-01, 5.15792370e-01],
...,
[-8.95374455e-03, -9.49803274e-03, -6.59012236e-03, ...,
-2.43885834e-02, -1.93168186e-02, 3.83974402e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
...,
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]]], dtype=float32):Rendering array...
,),}output: 384x0e
[[[-7.04744011e-02 1.21852076e+00 -9.89617705e-01 ... -9.95731533e-01
-7.08278120e-01 3.97574782e-01]
[ 6.35177255e-01 1.03016412e+00 -1.49667375e-02 ... -8.23826373e-01
3.75961483e-01 4.20294553e-01]
[-3.30404937e-02 4.40512866e-01 -1.09523706e-01 ... -4.86856103e-01
3.95714074e-01 5.15792191e-01]
...
[-8.95373151e-03 -9.49798059e-03 -6.59004319e-03 ... -2.43883822e-02
-1.93166938e-02 3.83972726e-03]
[-8.40501953e-03 -8.86404049e-03 -6.06027059e-03 ... -2.22131237e-02
-1.79946870e-02 3.61366617e-03]
[-7.22509157e-03 -7.52223935e-03 -4.98296646e-03 ... -1.78870130e-02
-1.52160246e-02 3.11819348e-03]]
[[-7.04744011e-02 1.21852076e+00 -9.89617705e-01 ... -9.95731533e-01
-7.08278120e-01 3.97574782e-01]
[ 6.35177970e-01 1.03016400e+00 -1.49669414e-02 ... -8.23826849e-01
3.75960767e-01 4.20294493e-01]
[-3.30404937e-02 4.40511018e-01 -1.09523691e-01 ... -4.86855090e-01
3.95713001e-01 5.15792251e-01]
...
[-8.95373151e-03 -9.49798059e-03 -6.59004319e-03 ... -2.43883822e-02
-1.93166938e-02 3.83972726e-03]
[-8.40497576e-03 -8.86400975e-03 -6.06032414e-03 ... -2.22131051e-02
-1.79946907e-02 3.61362263e-03]
[-7.22511625e-03 -7.52228592e-03 -4.98296879e-03 ... -1.78871062e-02
-1.52160879e-02 3.11822048e-03]]
[[-7.04736188e-02 1.21852052e+00 -9.89617407e-01 ... -9.95731711e-01
-7.08278000e-01 3.97574872e-01]
[ 6.35177970e-01 1.03016400e+00 -1.49669452e-02 ... -8.23826671e-01
3.75960618e-01 4.20294344e-01]
[-3.30405124e-02 4.40512449e-01 -1.09523349e-01 ... -4.86855447e-01
3.95714134e-01 5.15792370e-01]
...
[-8.95374455e-03 -9.49803274e-03 -6.59012236e-03 ... -2.43885834e-02
-1.93168186e-02 3.83974402e-03]
[-8.40501953e-03 -8.86404049e-03 -6.06027059e-03 ... -2.22131237e-02
-1.79946870e-02 3.61366617e-03]
[-7.22509157e-03 -7.52223935e-03 -4.98296646e-03 ... -1.78870130e-02
-1.52160246e-02 3.11819348e-03]]
...
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]]key=0,): Array([[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177255e-01, 1.03016412e+00, -1.49667375e-02, ...,
-8.23826373e-01, 3.75961483e-01, 4.20294553e-01],
[-3.30404937e-02, 4.40512866e-01, -1.09523706e-01, ...,
-4.86856103e-01, 3.95714074e-01, 5.15792191e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669414e-02, ...,
-8.23826849e-01, 3.75960767e-01, 4.20294493e-01],
[-3.30404937e-02, 4.40511018e-01, -1.09523691e-01, ...,
-4.86855090e-01, 3.95713001e-01, 5.15792251e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40497576e-03, -8.86400975e-03, -6.06032414e-03, ...,
-2.22131051e-02, -1.79946907e-02, 3.61362263e-03],
[-7.22511625e-03, -7.52228592e-03, -4.98296879e-03, ...,
-1.78871062e-02, -1.52160879e-02, 3.11822048e-03]],
[[-7.04736188e-02, 1.21852052e+00, -9.89617407e-01, ...,
-9.95731711e-01, -7.08278000e-01, 3.97574872e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669452e-02, ...,
-8.23826671e-01, 3.75960618e-01, 4.20294344e-01],
[-3.30405124e-02, 4.40512449e-01, -1.09523349e-01, ...,
-4.86855447e-01, 3.95714134e-01, 5.15792370e-01],
...,
[-8.95374455e-03, -9.49803274e-03, -6.59012236e-03, ...,
-2.43885834e-02, -1.93168186e-02, 3.83974402e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
...,
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]]], dtype=float32), :Rendering array...
,,),}output: 128x0e+128x1e+128x2e
[[-0.00081718 -0.04373479 0.04083708 ... -0.13225229 0.01714041
-0.00659576]
[-0.00081707 -0.0437349 0.04083706 ... 0.02510289 0.08776294
-0.09744485]
[-0.00081726 -0.04373484 0.04083706 ... -0.00060789 -0.10490344
-0.08260072]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00081718, -0.04373479, 0.04083708, ..., -0.13225229,
0.01714041, -0.00659576],
[-0.00081707, -0.0437349 , 0.04083706, ..., 0.02510289,
0.08776294, -0.09744485],
[-0.00081726, -0.04373484, 0.04083706, ..., -0.00060789,
-0.10490344, -0.08260072],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 128x0e+128x1e+128x2e
[[-0.00081718 -0.04373479 0.04083708 ... -0.13225229 0.01714041
-0.00659576]
[-0.00081707 -0.0437349 0.04083706 ... 0.02510289 0.08776294
-0.09744485]
[-0.00081726 -0.04373484 0.04083706 ... -0.00060789 -0.10490344
-0.08260072]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00081718, -0.04373479, 0.04083708, ..., -0.13225229,
0.01714041, -0.00659576],
[-0.00081707, -0.0437349 , 0.04083706, ..., 0.02510289,
0.08776294, -0.09744485],
[-0.00081726, -0.04373484, 0.04083706, ..., -0.00060789,
-0.10490344, -0.08260072],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 40960 parametersw[0,0] 128x0e,224x0e: Array([[ 8.8953382e-01, -9.6430862e-01, 1.4592638e+00, ...,
-6.1619239e+00, 3.9559295e+00, 1.8397622e+00],
[-5.0314933e-01, -1.6821375e+00, 2.0433755e+00, ...,
2.2306583e+00, 8.4645426e-01, -3.7939497e-04],
[-1.2256092e+00, 1.8113122e+00, -4.3403234e+00, ...,
-4.2426414e+00, -6.9003034e-01, 1.6837559e+00],
...,
[ 8.4209189e-02, -2.2738113e+00, -1.4586139e+00, ...,
-2.5382233e-01, -1.3814783e+00, 2.9294884e+00],
[-1.9275451e+00, -3.3198323e+00, -1.3414438e+00, ...,
5.2309602e-02, 3.8918704e-01, -4.7204347e+00],
[-9.1071028e-01, -1.0141186e+00, 9.5095253e-01, ...,
2.3143175e+00, 4.6380571e-01, -2.0695004e+00]], dtype=float32):Rendering array...
, w[1,1] 128x1e,64x1e: Array([[ 2.08344 , 2.0268428 , -0.10205896, ..., -0.6795716 ,
-0.9759364 , -2.71261 ],
[-1.3094982 , -1.9427985 , 1.2594589 , ..., -1.572477 ,
1.3153887 , 1.0310308 ],
[-0.09860689, 1.2901495 , 0.44501492, ..., 1.7403201 ,
-3.5873814 , 0.12359422],
...,
[ 2.3171134 , -0.4278286 , -2.723649 , ..., 2.31806 ,
4.9230413 , -3.0693502 ],
[ 1.5557939 , -2.9966815 , -2.6616237 , ..., -0.07412253,
-1.840733 , -0.17063624],
[ 1.3175749 , 0.1819717 , -0.4712196 , ..., -0.40548873,
1.1541386 , -1.3347238 ]], dtype=float32):Rendering array...
, w[2,2] 128x2e,32x2e: Array([[ 2.1402237 , -1.4459054 , -1.2280812 , ..., 0.5808771 ,
4.719111 , 0.23272462],
[ 0.5723715 , -2.2097077 , 0.5408782 , ..., 3.674021 ,
2.320326 , -0.91067314],
[ 1.716137 , 0.7318912 , 5.4104414 , ..., -0.8077885 ,
-0.01075178, -2.623032 ],
...,
[ 0.92815876, 1.0031501 , 5.834125 , ..., 1.7627509 ,
1.8706713 , -0.13664617],
[ 1.9301639 , -2.993204 , -5.7063103 , ..., 0.8726918 ,
0.07637297, -0.6783958 ],
[ 1.5137278 , -0.7742239 , -2.7060916 , ..., -1.565252 ,
3.3198183 , -1.4513925 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-3.4970281e-01 -1.6176869e-01 -4.7516149e-01 ... 1.1018757e-01
2.7865043e-02 -8.7888198e-05]
[-3.4970278e-01 -1.6176856e-01 -4.7516146e-01 ... 7.0680436e-03
-9.6611373e-02 5.9448101e-02]
[-3.4970289e-01 -1.6176876e-01 -4.7516170e-01 ... -3.4729265e-02
6.8746567e-02 8.3579808e-02]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-3.4970281e-01, -1.6176869e-01, -4.7516149e-01, ...,
1.1018757e-01, 2.7865043e-02, -8.7888198e-05],
[-3.4970278e-01, -1.6176856e-01, -4.7516146e-01, ...,
7.0680436e-03, -9.6611373e-02, 5.9448101e-02],
[-3.4970289e-01, -1.6176876e-01, -4.7516170e-01, ...,
-3.4729265e-02, 6.8746567e-02, 8.3579808e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-3.4970281e-01 -1.6176869e-01 -4.7516149e-01 ... 1.1018757e-01
2.7865043e-02 -8.7888198e-05]
[-3.4970278e-01 -1.6176856e-01 -4.7516146e-01 ... 7.0680436e-03
-9.6611373e-02 5.9448101e-02]
[-3.4970289e-01 -1.6176876e-01 -4.7516170e-01 ... -3.4729265e-02
6.8746567e-02 8.3579808e-02]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-3.4970281e-01, -1.6176869e-01, -4.7516149e-01, ...,
1.1018757e-01, 2.7865043e-02, -8.7888198e-05],
[-3.4970278e-01, -1.6176856e-01, -4.7516146e-01, ...,
7.0680436e-03, -9.6611373e-02, 5.9448101e-02],
[-3.4970289e-01, -1.6176876e-01, -4.7516170e-01, ...,
-3.4729265e-02, 6.8746567e-02, 8.3579808e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-1.7451271e-01 -8.9069709e-02 -3.4971526e-01 ... 1.1018757e-01
2.7865043e-02 -8.7888198e-05]
[-1.7451268e-01 -8.9069575e-02 -3.4971523e-01 ... 7.0680436e-03
-9.6611373e-02 5.9448101e-02]
[-1.7451280e-01 -8.9069784e-02 -3.4971547e-01 ... -3.4729265e-02
6.8746567e-02 8.3579808e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.7451271e-01, -8.9069709e-02, -3.4971526e-01, ...,
1.1018757e-01, 2.7865043e-02, -8.7888198e-05],
[-1.7451268e-01, -8.9069575e-02, -3.4971523e-01, ...,
7.0680436e-03, -9.6611373e-02, 5.9448101e-02],
[-1.7451280e-01, -8.9069784e-02, -3.4971547e-01, ...,
-3.4729265e-02, 6.8746567e-02, 8.3579808e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-1.7451271e-01 -8.9069709e-02 -3.4971526e-01 ... 1.1018757e-01
2.7865043e-02 -8.7888198e-05]
[-1.7451268e-01 -8.9069575e-02 -3.4971523e-01 ... 7.0680436e-03
-9.6611373e-02 5.9448101e-02]
[-1.7451280e-01 -8.9069784e-02 -3.4971547e-01 ... -3.4729265e-02
6.8746567e-02 8.3579808e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.7451271e-01, -8.9069709e-02, -3.4971526e-01, ...,
1.1018757e-01, 2.7865043e-02, -8.7888198e-05],
[-1.7451268e-01, -8.9069575e-02, -3.4971523e-01, ...,
7.0680436e-03, -9.6611373e-02, 5.9448101e-02],
[-1.7451280e-01, -8.9069784e-02, -3.4971547e-01, ...,
-3.4729265e-02, 6.8746567e-02, 8.3579808e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_1': layer_1input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.07596997, 0.27592677, -1.4187846 , ..., -3.4279747 ,
1.7853205 , 0.20697476],
[-0.5759112 , 0.37006366, -0.7312097 , ..., 0.05855327,
-1.455628 , 0.01047958],
[ 3.0478983 , 0.29383957, -1.444087 , ..., -2.9172626 ,
-2.151181 , 1.7908902 ],
...,
[ 1.0737331 , 1.1637673 , -1.0266234 , ..., 0.8998387 ,
0.19778317, -0.6234268 ],
[-0.60998833, 0.0140288 , 0.17401639, ..., -1.5538588 ,
2.0500894 , -0.22470145],
[ 0.97487503, 0.16422586, -1.996119 , ..., 1.6844265 ,
0.20047133, -1.0158596 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-1.8983171 , 1.1310133 , -1.0813652 , ..., 3.8598661 ,
2.1429558 , -3.254367 ],
[ 0.5758403 , 0.1438205 , -1.1729116 , ..., -0.40476117,
-1.7167196 , -0.3779419 ],
[-0.9746227 , 1.637837 , -0.05658302, ..., 0.01583426,
3.835662 , 2.9157832 ],
...,
[ 0.9743115 , 1.2317818 , 0.41286704, ..., 3.4449158 ,
-0.52608854, 0.23569065],
[ 0.68715185, -3.4482732 , 0.68070805, ..., -1.5814292 ,
-2.1625807 , -6.2308707 ],
[-1.7756596 , 2.637495 , -0.29488987, ..., 0.53851444,
1.3970186 , -6.2061367 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.4363009e-02, -4.7197115e-01, -1.3712782e+00, ...,
-1.2164356e+00, -1.1849614e+00, -6.1046684e-01],
[-7.4137789e-01, 3.9453822e-01, -1.5224578e+00, ...,
-6.9034678e-01, 1.0039064e+00, -1.2940993e+00],
[-6.5833054e-02, 1.0243961e+00, 1.2383620e+00, ...,
1.1483238e+00, 1.0708029e+00, 6.3221657e-01],
...,
[ 9.4188970e-01, -1.3368881e+00, 6.4393014e-02, ...,
-6.0346222e-01, -6.1310988e-02, -2.8410611e-01],
[-6.2800586e-01, 3.0434757e-01, 4.7315145e-01, ...,
-1.4627514e+00, -3.6217470e+00, 3.3898279e-03],
[ 3.7494395e+00, 3.9115545e-01, -8.2782209e-01, ...,
1.0386475e+00, -2.2443068e+00, 3.6211450e+00]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 4.2043066e-01 -1.8689492e+00 8.7586766e-01 ... -8.1766337e-02
2.1924494e-02 -1.6789339e-02]
[ 4.2043039e-01 -1.8689491e+00 8.7586677e-01 ... 1.5759172e-02
4.3092046e-02 -7.3095992e-02]
[ 4.2043057e-01 -1.8689487e+00 8.7586766e-01 ... -1.7127564e-02
-6.5016806e-02 -5.4108813e-02]
...
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 4.2043066e-01, -1.8689492e+00, 8.7586766e-01, ...,
-8.1766337e-02, 2.1924494e-02, -1.6789339e-02],
[ 4.2043039e-01, -1.8689491e+00, 8.7586677e-01, ...,
1.5759172e-02, 4.3092046e-02, -7.3095992e-02],
[ 4.2043057e-01, -1.8689487e+00, 8.7586766e-01, ...,
-1.7127564e-02, -6.5016806e-02, -5.4108813e-02],
...,
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-1.33558437e-01 -7.13425502e-02 -2.42414996e-01 ... 1.15343347e-01
2.91688740e-02 -9.20005696e-05]
[-1.33558407e-01 -7.13424385e-02 -2.42414966e-01 ... 7.39876926e-03
-1.01131976e-01 6.22297749e-02]
[-1.33558482e-01 -7.13426098e-02 -2.42415115e-01 ... -3.63542996e-02
7.19633251e-02 8.74906406e-02]
...
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -2.80259693e-45 2.80259693e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-1.33558437e-01, -7.13425502e-02, -2.42414996e-01, ...,
1.15343347e-01, 2.91688740e-02, -9.20005696e-05],
[-1.33558407e-01, -7.13424385e-02, -2.42414966e-01, ...,
7.39876926e-03, -1.01131976e-01, 6.22297749e-02],
[-1.33558482e-01, -7.13426098e-02, -2.42415115e-01, ...,
-3.63542996e-02, 7.19633251e-02, 8.74906406e-02],
...,
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -2.80259693e-45, 2.80259693e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[-0.4241752 , 1.5493343 , 1.1070297 , ..., 0.7761027 ,
0.97573304, -0.42901313],
[-1.5982518 , -0.762294 , 0.925268 , ..., 2.184838 ,
2.0235624 , -0.91705096],
[ 0.15138112, -2.1470785 , -2.1204956 , ..., 0.18950811,
1.1807846 , 0.4766244 ],
...,
[-0.1753623 , -0.1232187 , 1.0429176 , ..., 4.9537115 ,
-1.2018416 , -1.9440458 ],
[ 0.53979814, 0.6425637 , -1.3816024 , ..., -0.44571406,
0.14153725, 0.9647347 ],
[ 2.1930974 , -0.6061898 , -0.9843441 , ..., 2.3418922 ,
2.5735004 , -2.294102 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 1.3482215 , -3.131865 , -3.503224 , ..., 2.234558 ,
4.6431923 , -0.61837673],
[ 7.4621325 , -2.6024756 , -3.024557 , ..., 0.1146721 ,
-2.6666873 , -4.892023 ],
[-1.1567578 , 2.3914483 , -3.5104413 , ..., -2.923205 ,
-2.8599203 , 1.8045309 ],
...,
[ 3.134736 , -1.4929918 , 4.686314 , ..., -3.973341 ,
-8.705682 , 2.4547012 ],
[ 0.9738718 , -1.421875 , -1.2021968 , ..., 2.3651967 ,
-4.870255 , 1.5145608 ],
[ 3.0686798 , 4.702751 , -2.2753363 , ..., 2.1597495 ,
2.648089 , -0.3652111 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.3317432 , 1.6920732 , 0.44324556, ..., 4.037474 ,
-1.1616123 , 3.2893333 ],
[ 0.64936876, -3.7995343 , -0.47002053, ..., 3.9982092 ,
3.9305694 , 2.3129532 ],
[-1.3567086 , 6.0881386 , -0.23467521, ..., 2.5168328 ,
2.0815182 , 3.4210553 ],
...,
[-2.493092 , -2.0121279 , 1.7890359 , ..., -2.482454 ,
-2.4887905 , 6.641017 ],
[ 2.6620302 , 3.5397139 , -1.93738 , ..., -3.9678414 ,
-0.5936509 , -5.3279395 ],
[ 0.86180013, 2.0567636 , 1.5637033 , ..., 0.9133283 ,
0.8062022 , -1.2511878 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-2.1341591e+00 3.3727059e-01 -9.3139660e-01 ... -7.7589470e-01
1.2580271e-01 -1.1645695e-01]
[-2.1341588e+00 3.3727083e-01 -9.3139714e-01 ... 1.1569860e-01
4.6859056e-01 -6.3121963e-01]
[-2.1341588e+00 3.3727062e-01 -9.3139702e-01 ... -7.3006630e-02
-5.9439510e-01 -5.2227056e-01]
...
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.1341591e+00, 3.3727059e-01, -9.3139660e-01, ...,
-7.7589470e-01, 1.2580271e-01, -1.1645695e-01],
[-2.1341588e+00, 3.3727083e-01, -9.3139714e-01, ...,
1.1569860e-01, 4.6859056e-01, -6.3121963e-01],
[-2.1341588e+00, 3.3727062e-01, -9.3139702e-01, ...,
-7.3006630e-02, -5.9439510e-01, -5.2227056e-01],
...,
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.1341591e+00 3.3727059e-01 -9.3139660e-01 ... -7.7589470e-01
1.2580271e-01 -1.1645695e-01]
[-2.1341588e+00 3.3727083e-01 -9.3139714e-01 ... 1.1569860e-01
4.6859056e-01 -6.3121963e-01]
[-2.1341588e+00 3.3727062e-01 -9.3139702e-01 ... -7.3006630e-02
-5.9439510e-01 -5.2227056e-01]
...
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.9618179e-44 -7.0064923e-45 -5.6051939e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.1341591e+00, 3.3727059e-01, -9.3139660e-01, ...,
-7.7589470e-01, 1.2580271e-01, -1.1645695e-01],
[-2.1341588e+00, 3.3727083e-01, -9.3139714e-01, ...,
1.1569860e-01, 4.6859056e-01, -6.3121963e-01],
[-2.1341588e+00, 3.3727062e-01, -9.3139702e-01, ...,
-7.3006630e-02, -5.9439510e-01, -5.2227056e-01],
...,
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.9618179e-44, -7.0064923e-45, -5.6051939e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 4.052882 , -0.22530393, 1.4029229 , ..., 1.0817723 ,
-0.03070471, 1.6746305 ],
[ 5.135906 , -1.0228736 , 3.1861758 , ..., 2.607143 ,
4.6757793 , 4.5255265 ],
[ 2.4167147 , -0.24284168, 1.9592719 , ..., 1.1998506 ,
3.568256 , 3.412591 ],
...,
[ 0.69481957, -1.947504 , 2.3535092 , ..., -5.6909156 ,
1.383382 , 5.6895742 ],
[ 0.2564721 , -1.988878 , -0.11093269, ..., -3.8434715 ,
1.7036772 , 4.353121 ],
[-0.0734681 , -0.701135 , -0.6696881 , ..., -1.1655374 ,
-0.3284213 , 0.2726876 ]], dtype=float32):Rendering array...
,}output: Array([[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937311e+00, -1.02675724e+00, 6.91671014e-01, ...,
-4.61365968e-01, 1.07399237e+00, 1.51709604e+00],
[ 6.09271169e-01, 1.04536144e-02, 5.35353661e-01, ...,
-7.54954740e-02, -5.21018088e-01, 2.83327103e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937335e+00, -1.02675760e+00, 6.91670239e-01, ...,
-4.61365283e-01, 1.07399285e+00, 1.51709545e+00],
[ 6.09269798e-01, 1.04545997e-02, 5.35351455e-01, ...,
-7.54947066e-02, -5.21019697e-01, 2.83325016e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44897765e-02, -1.24929687e-02, -2.84479652e-03, ...,
-2.36891042e-02, -3.39861736e-02, 1.89452395e-02],
[ 1.20377662e-02, -1.00244759e-02, -2.40168930e-03, ...,
-1.91758256e-02, -2.87453085e-02, 1.41621530e-02]],
[[ 1.82870054e+00, -6.62221074e-01, -1.64561242e-01, ...,
1.97313941e+00, 1.62335014e+00, -5.23614772e-02],
[ 1.33937347e+00, -1.02675784e+00, 6.91670120e-01, ...,
-4.61365342e-01, 1.07399356e+00, 1.51709569e+00],
[ 6.09271049e-01, 1.04534263e-02, 5.35353482e-01, ...,
-7.54962489e-02, -5.21018147e-01, 2.83327669e-01],
...,
[ 1.56872347e-02, -1.37338489e-02, -3.05157690e-03, ...,
-2.59463526e-02, -3.64794694e-02, 2.14571767e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
...,
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 1.82870078e+00 -6.62220716e-01 -1.64561242e-01 ... 1.97314012e+00
1.62335014e+00 -5.23620546e-02]
[ 1.33937311e+00 -1.02675724e+00 6.91671014e-01 ... -4.61365968e-01
1.07399237e+00 1.51709604e+00]
[ 6.09271169e-01 1.04536144e-02 5.35353661e-01 ... -7.54954740e-02
-5.21018088e-01 2.83327103e-01]
...
[ 1.56871751e-02 -1.37338126e-02 -3.05151008e-03 ... -2.59463191e-02
-3.64792608e-02 2.14571245e-02]
[ 1.44898985e-02 -1.24931475e-02 -2.84474064e-03 ... -2.36894805e-02
-3.39862257e-02 1.89456847e-02]
[ 1.20376768e-02 -1.00243539e-02 -2.40167952e-03 ... -1.91756040e-02
-2.87451912e-02 1.41618997e-02]]
[[ 1.82870078e+00 -6.62220716e-01 -1.64561242e-01 ... 1.97314012e+00
1.62335014e+00 -5.23620546e-02]
[ 1.33937335e+00 -1.02675760e+00 6.91670239e-01 ... -4.61365283e-01
1.07399285e+00 1.51709545e+00]
[ 6.09269798e-01 1.04545997e-02 5.35351455e-01 ... -7.54947066e-02
-5.21019697e-01 2.83325016e-01]
...
[ 1.56871751e-02 -1.37338126e-02 -3.05151008e-03 ... -2.59463191e-02
-3.64792608e-02 2.14571245e-02]
[ 1.44897765e-02 -1.24929687e-02 -2.84479652e-03 ... -2.36891042e-02
-3.39861736e-02 1.89452395e-02]
[ 1.20377662e-02 -1.00244759e-02 -2.40168930e-03 ... -1.91758256e-02
-2.87453085e-02 1.41621530e-02]]
[[ 1.82870054e+00 -6.62221074e-01 -1.64561242e-01 ... 1.97313941e+00
1.62335014e+00 -5.23614772e-02]
[ 1.33937347e+00 -1.02675784e+00 6.91670120e-01 ... -4.61365342e-01
1.07399356e+00 1.51709569e+00]
[ 6.09271049e-01 1.04534263e-02 5.35353482e-01 ... -7.54962489e-02
-5.21018147e-01 2.83327669e-01]
...
[ 1.56872347e-02 -1.37338489e-02 -3.05157690e-03 ... -2.59463526e-02
-3.64794694e-02 2.14571767e-02]
[ 1.44898985e-02 -1.24931475e-02 -2.84474064e-03 ... -2.36894805e-02
-3.39862257e-02 1.89456847e-02]
[ 1.20376768e-02 -1.00243539e-02 -2.40167952e-03 ... -1.91756040e-02
-2.87451912e-02 1.41618997e-02]]
...
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]]key=0,): Array([[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937311e+00, -1.02675724e+00, 6.91671014e-01, ...,
-4.61365968e-01, 1.07399237e+00, 1.51709604e+00],
[ 6.09271169e-01, 1.04536144e-02, 5.35353661e-01, ...,
-7.54954740e-02, -5.21018088e-01, 2.83327103e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937335e+00, -1.02675760e+00, 6.91670239e-01, ...,
-4.61365283e-01, 1.07399285e+00, 1.51709545e+00],
[ 6.09269798e-01, 1.04545997e-02, 5.35351455e-01, ...,
-7.54947066e-02, -5.21019697e-01, 2.83325016e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44897765e-02, -1.24929687e-02, -2.84479652e-03, ...,
-2.36891042e-02, -3.39861736e-02, 1.89452395e-02],
[ 1.20377662e-02, -1.00244759e-02, -2.40168930e-03, ...,
-1.91758256e-02, -2.87453085e-02, 1.41621530e-02]],
[[ 1.82870054e+00, -6.62221074e-01, -1.64561242e-01, ...,
1.97313941e+00, 1.62335014e+00, -5.23614772e-02],
[ 1.33937347e+00, -1.02675784e+00, 6.91670120e-01, ...,
-4.61365342e-01, 1.07399356e+00, 1.51709569e+00],
[ 6.09271049e-01, 1.04534263e-02, 5.35353482e-01, ...,
-7.54962489e-02, -5.21018147e-01, 2.83327669e-01],
...,
[ 1.56872347e-02, -1.37338489e-02, -3.05157690e-03, ...,
-2.59463526e-02, -3.64794694e-02, 2.14571767e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
...,
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[-0.25765935 0.05459043 -0.02816507 ... 0.00181926 -0.01426247
0.00675845]
[-0.2576594 0.05459045 -0.02816471 ... -0.00653084 0.00870601
0.01158378]
[-0.25765952 0.05459048 -0.028165 ... 0.01485612 0.00555922
-0.00076269]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.25765935, 0.05459043, -0.02816507, ..., 0.00181926,
-0.01426247, 0.00675845],
[-0.2576594 , 0.05459045, -0.02816471, ..., -0.00653084,
0.00870601, 0.01158378],
[-0.25765952, 0.05459048, -0.028165 , ..., 0.01485612,
0.00555922, -0.00076269],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[-0.25765935 0.05459043 -0.02816507 ... 0.00181926 -0.01426247
0.00675845]
[-0.2576594 0.05459045 -0.02816471 ... -0.00653084 0.00870601
0.01158378]
[-0.25765952 0.05459048 -0.028165 ... 0.01485612 0.00555922
-0.00076269]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.25765935, 0.05459043, -0.02816507, ..., 0.00181926,
-0.01426247, 0.00675845],
[-0.2576594 , 0.05459045, -0.02816471, ..., -0.00653084,
0.00870601, 0.01158378],
[-0.25765952, 0.05459048, -0.028165 , ..., 0.01485612,
0.00555922, -0.00076269],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 3.6113713 , -1.1181697 , 0.03881168, ..., -2.1805305 ,
0.64018303, 1.2658951 ],
[-3.7488034 , -0.92508084, -1.101863 , ..., 0.8667244 ,
-3.71324 , -0.2927172 ],
[ 1.1253558 , 1.2100322 , 0.13155553, ..., -2.042996 ,
-2.523338 , 3.6498551 ],
...,
[ 3.6797526 , 3.550883 , 1.4122659 , ..., 2.826467 ,
1.8865697 , -1.0979416 ],
[ 0.4086895 , 0.00904947, -0.63308686, ..., -0.81488013,
3.6393635 , -3.7455666 ],
[ 1.8093982 , -3.068014 , 0.901131 , ..., -1.4847515 ,
-0.36652914, 1.6428287 ]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[-0.79561204, -0.30382457, 0.0335411 , ..., -1.2693908 ,
0.37836167, 2.9927666 ],
[-1.455988 , -1.9610479 , -0.44855782, ..., 0.03941472,
0.62473786, 4.7411556 ],
[ 0.7098443 , 0.6184685 , -2.0183759 , ..., -5.9243107 ,
-4.4614496 , 2.7931871 ],
...,
[-0.7004779 , 1.4356883 , -0.9977043 , ..., -0.51720756,
0.15333058, -1.0994623 ],
[ 0.77050114, 1.6966777 , -1.5533303 , ..., -1.0010715 ,
-0.4932081 , 0.49124607],
[-1.25783 , -0.6229767 , 0.94102746, ..., -0.6150069 ,
0.45159495, 0.16357416]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[ 1.9871085 , -3.3893056 , -0.04077484, ..., 0.21905695,
0.7577846 , -0.5522776 ],
[ -0.5850577 , 0.19033383, -2.503004 , ..., 1.1635637 ,
1.7451153 , -3.0206232 ],
[ 1.2223674 , -1.2091054 , 0.39926517, ..., 2.9988463 ,
-1.7507242 , -3.1746583 ],
...,
[ 0.5759754 , -0.16012855, 3.1206794 , ..., 3.6077344 ,
-11.050401 , 0.70109856],
[ 2.6171324 , 1.2029477 , 5.376402 , ..., 0.8496376 ,
-0.45670396, 1.9353403 ],
[ -0.88361245, 5.8734903 , -1.6447512 , ..., 3.8160083 ,
2.0072403 , -1.1579607 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-0.8230932 -1.3913971 -1.6462318 ... -0.03879504 -0.07261174
0.06403933]
[-0.8230936 -1.3913981 -1.6462328 ... -0.00805056 0.09310944
0.04628373]
[-0.8230937 -1.3913968 -1.6462322 ... 0.10093144 -0.02048295
-0.0166152 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.8230932 , -1.3913971 , -1.6462318 , ..., -0.03879504,
-0.07261174, 0.06403933],
[-0.8230936 , -1.3913981 , -1.6462328 , ..., -0.00805056,
0.09310944, 0.04628373],
[-0.8230937 , -1.3913968 , -1.6462322 , ..., 0.10093144,
-0.02048295, -0.0166152 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-0.8230932 -1.3913971 -1.6462318 ... -0.03879504 -0.07261174
0.06403933]
[-0.8230936 -1.3913981 -1.6462328 ... -0.00805056 0.09310944
0.04628373]
[-0.8230937 -1.3913968 -1.6462322 ... 0.10093144 -0.02048295
-0.0166152 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.8230932 , -1.3913971 , -1.6462318 , ..., -0.03879504,
-0.07261174, 0.06403933],
[-0.8230936 , -1.3913981 , -1.6462328 , ..., -0.00805056,
0.09310944, 0.04628373],
[-0.8230937 , -1.3913968 , -1.6462322 , ..., 0.10093144,
-0.02048295, -0.0166152 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-4.02662516e-01 -3.26034641e+00 -7.70364106e-01 ... -1.20561376e-01
-5.06872460e-02 4.72499877e-02]
[-4.02663201e-01 -3.26034713e+00 -7.70366073e-01 ... 7.70861655e-03
1.36201486e-01 -2.68122666e-02]
[-4.02663141e-01 -3.26034546e+00 -7.70364583e-01 ... 8.38038698e-02
-8.54997560e-02 -7.07240179e-02]
...
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.02662516e-01, -3.26034641e+00, -7.70364106e-01, ...,
-1.20561376e-01, -5.06872460e-02, 4.72499877e-02],
[-4.02663201e-01, -3.26034713e+00, -7.70366073e-01, ...,
7.70861655e-03, 1.36201486e-01, -2.68122666e-02],
[-4.02663141e-01, -3.26034546e+00, -7.70364583e-01, ...,
8.38038698e-02, -8.54997560e-02, -7.07240179e-02],
...,
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-4.02662516e-01 -3.26034641e+00 -7.70364106e-01 ... -1.20561376e-01
-5.06872460e-02 4.72499877e-02]
[-4.02663201e-01 -3.26034713e+00 -7.70366073e-01 ... 7.70861655e-03
1.36201486e-01 -2.68122666e-02]
[-4.02663141e-01 -3.26034546e+00 -7.70364583e-01 ... 8.38038698e-02
-8.54997560e-02 -7.07240179e-02]
...
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-5.60519386e-45 9.80908925e-45 -1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.02662516e-01, -3.26034641e+00, -7.70364106e-01, ...,
-1.20561376e-01, -5.06872460e-02, 4.72499877e-02],
[-4.02663201e-01, -3.26034713e+00, -7.70366073e-01, ...,
7.70861655e-03, 1.36201486e-01, -2.68122666e-02],
[-4.02663141e-01, -3.26034546e+00, -7.70364583e-01, ...,
8.38038698e-02, -8.54997560e-02, -7.07240179e-02],
...,
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-5.60519386e-45, 9.80908925e-45, -1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_2': layer_2input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.2698589 , 0.44674757, -0.31250012, ..., -3.6976795 ,
-2.4428482 , -1.1022315 ],
[-0.19646236, -0.9834637 , 0.63143486, ..., -0.05126027,
1.1076013 , 1.7861416 ],
[ 0.48316053, 2.9596577 , 0.12952845, ..., -0.56942916,
-0.40692878, -0.22096917],
...,
[-3.9606133 , 0.03599317, 1.4394854 , ..., 0.19181603,
-1.2654991 , -1.2876765 ],
[-2.0281868 , -0.13632914, -3.4674752 , ..., -0.2869467 ,
-0.17545098, -1.5250418 ],
[ 4.2003765 , 0.12415113, -0.42462096, ..., 2.0245545 ,
-2.225135 , 2.1616156 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 2.2523465 , -1.2124305 , 0.4407775 , ..., -0.11525499,
0.6420827 , -1.1333939 ],
[-2.3259945 , -2.6467135 , -0.9030307 , ..., -3.7320697 ,
2.6729276 , -3.3454947 ],
[ 3.0255785 , 0.5180181 , 2.7732878 , ..., 1.524265 ,
-2.1999319 , -0.20321423],
...,
[-0.57588875, -0.53166825, 0.2651183 , ..., 2.953936 ,
-1.5758833 , -2.0478845 ],
[-2.9871452 , 5.09833 , -0.8852749 , ..., -1.3664873 ,
1.2529833 , 1.8465164 ],
[-4.4754567 , 3.4820943 , -2.7085273 , ..., -2.4177372 ,
-0.8317587 , -0.06291667]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-0.42950523, 0.9804102 , -0.34976012, ..., -3.4437647 ,
-3.3836908 , -0.8078704 ],
[-1.6786072 , -1.2309582 , 0.22875902, ..., -0.72541004,
1.2473128 , 0.3433992 ],
[-0.87624604, 1.8254704 , -3.3011746 , ..., -1.3931991 ,
-1.4904697 , 2.3584936 ],
...,
[ 1.6216624 , -1.4492549 , -0.44740498, ..., 0.48102644,
-1.2849988 , 0.5881696 ],
[-0.59220976, -0.5082739 , -1.5561093 , ..., -0.5740662 ,
-0.3018164 , 1.8806973 ],
[-1.3057946 , -0.49188724, 0.6209479 , ..., -0.8584203 ,
-0.4544655 , 0.9636952 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-9.51450169e-01 -1.91375899e+00 -1.05446374e+00 ... 7.87642002e-02
-4.04610932e-02 -1.42762270e-02]
[-9.51449215e-01 -1.91376054e+00 -1.05446398e+00 ... -4.94634733e-02
-4.50182967e-02 5.97480200e-02]
[-9.51450586e-01 -1.91375899e+00 -1.05446374e+00 ... 1.12172095e-02
8.54915902e-02 2.47158632e-02]
...
[ 5.60519386e-45 1.40129846e-45 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 5.60519386e-45 1.40129846e-45 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 5.60519386e-45 1.40129846e-45 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-9.51450169e-01, -1.91375899e+00, -1.05446374e+00, ...,
7.87642002e-02, -4.04610932e-02, -1.42762270e-02],
[-9.51449215e-01, -1.91376054e+00, -1.05446398e+00, ...,
-4.94634733e-02, -4.50182967e-02, 5.97480200e-02],
[-9.51450586e-01, -1.91375899e+00, -1.05446374e+00, ...,
1.12172095e-02, 8.54915902e-02, 2.47158632e-02],
...,
[ 5.60519386e-45, 1.40129846e-45, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 5.60519386e-45, 1.40129846e-45, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 5.60519386e-45, 1.40129846e-45, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-2.7049014e-01 -2.0201305e-01 -4.0865174e-01 ... 3.7446331e-02
1.5743446e-02 -1.4675833e-02]
[-2.7049047e-01 -2.0201294e-01 -4.0865225e-01 ... -2.3942911e-03
-4.2304091e-02 8.3278716e-03]
[-2.7049047e-01 -2.0201313e-01 -4.0865186e-01 ... -2.6029449e-02
2.6556190e-02 2.1966854e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7049014e-01, -2.0201305e-01, -4.0865174e-01, ...,
3.7446331e-02, 1.5743446e-02, -1.4675833e-02],
[-2.7049047e-01, -2.0201294e-01, -4.0865225e-01, ...,
-2.3942911e-03, -4.2304091e-02, 8.3278716e-03],
[-2.7049047e-01, -2.0201313e-01, -4.0865186e-01, ...,
-2.6029449e-02, 2.6556190e-02, 2.1966854e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[-2.1586008 , -1.6786722 , 0.5165552 , ..., 0.70764744,
-3.3874962 , -0.5979072 ],
[ 1.4567249 , -0.78757703, 0.14725283, ..., 3.9408135 ,
-2.4566958 , 2.719328 ],
[ 1.5687432 , -1.134582 , 0.3726073 , ..., -0.5147848 ,
-2.6269054 , 1.254605 ],
...,
[-4.5462236 , -2.068843 , -0.07038818, ..., -2.0033019 ,
0.72902334, -1.1020043 ],
[ 0.6327289 , -3.9143023 , 0.30573696, ..., -1.8584297 ,
2.3847718 , 1.774803 ],
[ 0.5761787 , 2.6509273 , 1.4255936 , ..., 1.262279 ,
2.7701647 , -2.3315237 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 0.6084777 , -0.66246223, 6.619954 , ..., -0.9813953 ,
1.444019 , 1.6443533 ],
[-1.2975378 , 2.1629822 , -2.5576208 , ..., -2.1381264 ,
0.77533215, -1.0800792 ],
[ 1.0625116 , -2.1124494 , -0.28465724, ..., -0.13889353,
-2.429628 , -3.6430233 ],
...,
[-0.583007 , 3.0251534 , 3.7068667 , ..., 2.6721835 ,
1.8640645 , 3.0045564 ],
[ 2.132253 , 5.790305 , 1.7161772 , ..., -1.8887897 ,
1.0866327 , 0.86480767],
[-0.6644205 , 4.505556 , 2.7787488 , ..., -0.5326051 ,
3.594365 , -9.192757 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-4.14843 , -2.5014877 , -1.4684834 , ..., -5.8136964 ,
0.8666029 , 3.7350478 ],
[-3.7453384 , 1.0307775 , -3.6572578 , ..., 1.1598345 ,
-0.38757676, 0.70694864],
[-1.8403573 , 0.3684081 , 0.52219 , ..., -1.808346 ,
-0.20554003, 2.7274349 ],
...,
[-0.22899586, 5.4061036 , 2.9266653 , ..., 1.767967 ,
3.704977 , -4.4892683 ],
[-0.06069649, 3.433467 , -3.772579 , ..., 1.0408311 ,
0.47197062, -0.97904927],
[ 5.2918415 , -5.065796 , -5.8402925 , ..., -2.3323772 ,
-2.5990508 , 7.3229866 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[ 7.9026639e-01 3.0368912e-01 1.7638429e+00 ... 1.6508137e-01
-4.7206515e-01 1.7014803e-01]
[ 7.9026711e-01 3.0368870e-01 1.7638451e+00 ... -2.6422718e-01
1.8588649e-01 4.1801792e-01]
[ 7.9026610e-01 3.0368882e-01 1.7638420e+00 ... 4.4390211e-01
2.8613928e-01 9.0990812e-03]
...
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9026639e-01, 3.0368912e-01, 1.7638429e+00, ...,
1.6508137e-01, -4.7206515e-01, 1.7014803e-01],
[ 7.9026711e-01, 3.0368870e-01, 1.7638451e+00, ...,
-2.6422718e-01, 1.8588649e-01, 4.1801792e-01],
[ 7.9026610e-01, 3.0368882e-01, 1.7638420e+00, ...,
4.4390211e-01, 2.8613928e-01, 9.0990812e-03],
...,
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 7.9026639e-01 3.0368912e-01 1.7638429e+00 ... 1.6508137e-01
-4.7206515e-01 1.7014803e-01]
[ 7.9026711e-01 3.0368870e-01 1.7638451e+00 ... -2.6422718e-01
1.8588649e-01 4.1801792e-01]
[ 7.9026610e-01 3.0368882e-01 1.7638420e+00 ... 4.4390211e-01
2.8613928e-01 9.0990812e-03]
...
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9026639e-01, 3.0368912e-01, 1.7638429e+00, ...,
1.6508137e-01, -4.7206515e-01, 1.7014803e-01],
[ 7.9026711e-01, 3.0368870e-01, 1.7638451e+00, ...,
-2.6422718e-01, 1.8588649e-01, 4.1801792e-01],
[ 7.9026610e-01, 3.0368882e-01, 1.7638420e+00, ...,
4.4390211e-01, 2.8613928e-01, 9.0990812e-03],
...,
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 1.9611268 , 0.89812714, -1.9948477 , ..., 0.30886963,
1.1508706 , 3.4620895 ],
[ 1.8762344 , 0.6253016 , -1.980058 , ..., -0.70826226,
1.139967 , -0.7580052 ],
[-1.3364033 , 0.55748296, 0.44204178, ..., -3.708407 ,
0.45144904, 0.67026114],
...,
[ 0.10644482, 0.4301311 , 0.04070838, ..., 1.2164056 ,
-4.283859 , 3.9550807 ],
[-0.40197 , 0.56505716, 0.7024608 , ..., -0.2792783 ,
-5.6710806 , 0.0624095 ],
[-0.3675052 , 0.54707265, 0.21580404, ..., -0.57510746,
-3.7059128 , -0.69469917]], dtype=float32):Rendering array...
,}output: Array([[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315867e+00, 1.99174806e-01, -8.51171792e-01, ...,
2.99210519e-01, 7.56071627e-01, 5.14216542e-01],
[ 6.54132724e-01, -1.76375911e-01, -5.85293889e-01, ...,
9.83544767e-01, 6.02065086e-01, 4.26705986e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315879e+00, 1.99175254e-01, -8.51171911e-01, ...,
2.99209505e-01, 7.56070971e-01, 5.14215589e-01],
[ 6.54131353e-01, -1.76375523e-01, -5.85292637e-01, ...,
9.83543694e-01, 6.02060676e-01, 4.26704019e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92134960e-03, -1.47916619e-02, -4.25949041e-03, ...,
-7.94653222e-03, 6.34376779e-02, 1.89574063e-02],
[-2.40497058e-03, -1.15050329e-02, -3.78032262e-03, ...,
-6.95147086e-03, 5.09458557e-02, 1.65958572e-02]],
[[ 7.35480726e-01, 7.03230560e-01, -8.26478004e-01, ...,
-1.49849820e+00, 8.28793943e-01, -1.00703466e+00],
[ 1.02315843e+00, 1.99175462e-01, -8.51171792e-01, ...,
2.99208820e-01, 7.56070375e-01, 5.14215410e-01],
[ 6.54132426e-01, -1.76375672e-01, -5.85293651e-01, ...,
9.83544528e-01, 6.02063060e-01, 4.26706076e-01],
...,
[-3.17242346e-03, -1.64807681e-02, -4.46723634e-03, ...,
-8.39057192e-03, 6.97437152e-02, 2.00373363e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
...,
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 7.35480607e-01 7.03230560e-01 -8.26477945e-01 ... -1.49849844e+00
8.28794420e-01 -1.00703466e+00]
[ 1.02315867e+00 1.99174806e-01 -8.51171792e-01 ... 2.99210519e-01
7.56071627e-01 5.14216542e-01]
[ 6.54132724e-01 -1.76375911e-01 -5.85293889e-01 ... 9.83544767e-01
6.02065086e-01 4.26705986e-01]
...
[-3.17240926e-03 -1.64806414e-02 -4.46722098e-03 ... -8.39055143e-03
6.97432309e-02 2.00373754e-02]
[-2.92137428e-03 -1.47917420e-02 -4.25948715e-03 ... -7.94654153e-03
6.34378195e-02 1.89575516e-02]
[-2.40494753e-03 -1.15049435e-02 -3.78031936e-03 ... -6.95143500e-03
5.09456061e-02 1.65958032e-02]]
[[ 7.35480607e-01 7.03230560e-01 -8.26477945e-01 ... -1.49849844e+00
8.28794420e-01 -1.00703466e+00]
[ 1.02315879e+00 1.99175254e-01 -8.51171911e-01 ... 2.99209505e-01
7.56070971e-01 5.14215589e-01]
[ 6.54131353e-01 -1.76375523e-01 -5.85292637e-01 ... 9.83543694e-01
6.02060676e-01 4.26704019e-01]
...
[-3.17240926e-03 -1.64806414e-02 -4.46722098e-03 ... -8.39055143e-03
6.97432309e-02 2.00373754e-02]
[-2.92134960e-03 -1.47916619e-02 -4.25949041e-03 ... -7.94653222e-03
6.34376779e-02 1.89574063e-02]
[-2.40497058e-03 -1.15050329e-02 -3.78032262e-03 ... -6.95147086e-03
5.09458557e-02 1.65958572e-02]]
[[ 7.35480726e-01 7.03230560e-01 -8.26478004e-01 ... -1.49849820e+00
8.28793943e-01 -1.00703466e+00]
[ 1.02315843e+00 1.99175462e-01 -8.51171792e-01 ... 2.99208820e-01
7.56070375e-01 5.14215410e-01]
[ 6.54132426e-01 -1.76375672e-01 -5.85293651e-01 ... 9.83544528e-01
6.02063060e-01 4.26706076e-01]
...
[-3.17242346e-03 -1.64807681e-02 -4.46723634e-03 ... -8.39057192e-03
6.97437152e-02 2.00373363e-02]
[-2.92137428e-03 -1.47917420e-02 -4.25948715e-03 ... -7.94654153e-03
6.34378195e-02 1.89575516e-02]
[-2.40494753e-03 -1.15049435e-02 -3.78031936e-03 ... -6.95143500e-03
5.09456061e-02 1.65958032e-02]]
...
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]]key=0,): Array([[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315867e+00, 1.99174806e-01, -8.51171792e-01, ...,
2.99210519e-01, 7.56071627e-01, 5.14216542e-01],
[ 6.54132724e-01, -1.76375911e-01, -5.85293889e-01, ...,
9.83544767e-01, 6.02065086e-01, 4.26705986e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315879e+00, 1.99175254e-01, -8.51171911e-01, ...,
2.99209505e-01, 7.56070971e-01, 5.14215589e-01],
[ 6.54131353e-01, -1.76375523e-01, -5.85292637e-01, ...,
9.83543694e-01, 6.02060676e-01, 4.26704019e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92134960e-03, -1.47916619e-02, -4.25949041e-03, ...,
-7.94653222e-03, 6.34376779e-02, 1.89574063e-02],
[-2.40497058e-03, -1.15050329e-02, -3.78032262e-03, ...,
-6.95147086e-03, 5.09458557e-02, 1.65958572e-02]],
[[ 7.35480726e-01, 7.03230560e-01, -8.26478004e-01, ...,
-1.49849820e+00, 8.28793943e-01, -1.00703466e+00],
[ 1.02315843e+00, 1.99175462e-01, -8.51171792e-01, ...,
2.99208820e-01, 7.56070375e-01, 5.14215410e-01],
[ 6.54132426e-01, -1.76375672e-01, -5.85293651e-01, ...,
9.83544528e-01, 6.02063060e-01, 4.26706076e-01],
...,
[-3.17242346e-03, -1.64807681e-02, -4.46723634e-03, ...,
-8.39057192e-03, 6.97437152e-02, 2.00373363e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
...,
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[-0.10364437 0.03685197 -0.30450645 ... 0.10318621 -0.0372932
-0.04058116]
[-0.1036446 0.03685193 -0.3045061 ... -0.06716537 -0.07632063
0.05781792]
[-0.10364437 0.03685208 -0.3045065 ... -0.01129957 0.11360317
0.02550665]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.10364437, 0.03685197, -0.30450645, ..., 0.10318621,
-0.0372932 , -0.04058116],
[-0.1036446 , 0.03685193, -0.3045061 , ..., -0.06716537,
-0.07632063, 0.05781792],
[-0.10364437, 0.03685208, -0.3045065 , ..., -0.01129957,
0.11360317, 0.02550665],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[-0.10364437 0.03685197 -0.30450645 ... 0.10318621 -0.0372932
-0.04058116]
[-0.1036446 0.03685193 -0.3045061 ... -0.06716537 -0.07632063
0.05781792]
[-0.10364437 0.03685208 -0.3045065 ... -0.01129957 0.11360317
0.02550665]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.10364437, 0.03685197, -0.30450645, ..., 0.10318621,
-0.0372932 , -0.04058116],
[-0.1036446 , 0.03685193, -0.3045061 , ..., -0.06716537,
-0.07632063, 0.05781792],
[-0.10364437, 0.03685208, -0.3045065 , ..., -0.01129957,
0.11360317, 0.02550665],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 2.0081546 , 1.6479783 , 1.5604763 , ..., 1.4455466 ,
-1.0169146 , -1.3716726 ],
[-2.3814116 , -0.9727504 , -1.6676872 , ..., 0.1920512 ,
-0.51205426, -2.6818907 ],
[ 1.6531345 , 1.0792857 , -0.8966044 , ..., -1.891223 ,
-1.4565222 , 1.7349514 ],
...,
[ 2.9895694 , 2.408599 , -0.29375437, ..., -1.8559114 ,
5.440316 , -5.4067397 ],
[ 1.5446353 , -0.12033591, 1.850207 , ..., -0.9702179 ,
1.4745461 , 3.9028435 ],
[-2.7847724 , -2.2506235 , -1.7330496 , ..., 5.0526257 ,
2.3648324 , 0.84812665]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[-4.5358515 , 1.1497406 , 2.589901 , ..., -2.043665 ,
0.22063464, 1.3551856 ],
[-1.0841694 , -1.8803372 , 0.6187718 , ..., 3.6525974 ,
-2.4811273 , 2.0372815 ],
[-4.234338 , 2.2495852 , -3.7499642 , ..., -3.6261945 ,
2.5682158 , -0.564426 ],
...,
[ 1.5536867 , -0.52748156, -1.7037528 , ..., -1.7256256 ,
0.5423395 , 0.8801775 ],
[ 0.59128857, -0.13331643, -0.565452 , ..., 2.2760124 ,
-2.2967505 , 1.0503368 ],
[-0.9111213 , -1.1474068 , -0.18583015, ..., -0.9213156 ,
2.2737846 , 0.03138326]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[-1.3098822 , -4.5079107 , -1.8133179 , ..., -0.64720494,
2.876182 , 1.0001799 ],
[-3.1235614 , 2.1366012 , -1.4442047 , ..., -1.5842416 ,
-0.0381077 , 5.034386 ],
[ 1.4682164 , -1.0672374 , 2.318073 , ..., 1.0078801 ,
-0.9195423 , 1.4017334 ],
...,
[ 2.1997006 , -1.3968203 , -1.3433715 , ..., 2.8820744 ,
-0.2825411 , 0.8039047 ],
[-1.2277429 , 0.2927603 , 4.215725 , ..., -2.0764709 ,
3.7434182 , -5.0725036 ],
[-2.206367 , -6.099845 , -1.0706434 , ..., 4.453828 ,
-0.0274033 , -0.7044329 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 0.09951255 -0.19077517 -0.35172096 ... 0.01510658 -0.20430966
-0.00644494]
[ 0.09955537 -0.19076395 -0.35170037 ... -0.15937763 0.08793367
0.09426563]
[ 0.09955747 -0.19075295 -0.3517181 ... 0.14708759 0.11618464
-0.08270775]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.09951255, -0.19077517, -0.35172096, ..., 0.01510658,
-0.20430966, -0.00644494],
[ 0.09955537, -0.19076395, -0.35170037, ..., -0.15937763,
0.08793367, 0.09426563],
[ 0.09955747, -0.19075295, -0.3517181 , ..., 0.14708759,
0.11618464, -0.08270775],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[ 0.09951255 -0.19077517 -0.35172096 ... 0.01510658 -0.20430966
-0.00644494]
[ 0.09955537 -0.19076395 -0.35170037 ... -0.15937763 0.08793367
0.09426563]
[ 0.09955747 -0.19075295 -0.3517181 ... 0.14708759 0.11618464
-0.08270775]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.09951255, -0.19077517, -0.35172096, ..., 0.01510658,
-0.20430966, -0.00644494],
[ 0.09955537, -0.19076395, -0.35170037, ..., -0.15937763,
0.08793367, 0.09426563],
[ 0.09955747, -0.19075295, -0.3517181 , ..., 0.14708759,
0.11618464, -0.08270775],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-8.5193765e-01 -2.1045341e+00 -1.4061847e+00 ... 9.3870781e-02
-2.4477075e-01 -2.0721167e-02]
[-8.5189384e-01 -2.1045246e+00 -1.4061644e+00 ... -2.0884112e-01
4.2915378e-02 1.5401365e-01]
[-8.5189313e-01 -2.1045120e+00 -1.4061818e+00 ... 1.5830480e-01
2.0167623e-01 -5.7991885e-02]
...
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.5193765e-01, -2.1045341e+00, -1.4061847e+00, ...,
9.3870781e-02, -2.4477075e-01, -2.0721167e-02],
[-8.5189384e-01, -2.1045246e+00, -1.4061644e+00, ...,
-2.0884112e-01, 4.2915378e-02, 1.5401365e-01],
[-8.5189313e-01, -2.1045120e+00, -1.4061818e+00, ...,
1.5830480e-01, 2.0167623e-01, -5.7991885e-02],
...,
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-8.5193765e-01 -2.1045341e+00 -1.4061847e+00 ... 9.3870781e-02
-2.4477075e-01 -2.0721167e-02]
[-8.5189384e-01 -2.1045246e+00 -1.4061644e+00 ... -2.0884112e-01
4.2915378e-02 1.5401365e-01]
[-8.5189313e-01 -2.1045120e+00 -1.4061818e+00 ... 1.5830480e-01
2.0167623e-01 -5.7991885e-02]
...
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.5193765e-01, -2.1045341e+00, -1.4061847e+00, ...,
9.3870781e-02, -2.4477075e-01, -2.0721167e-02],
[-8.5189384e-01, -2.1045246e+00, -1.4061644e+00, ...,
-2.0884112e-01, 4.2915378e-02, 1.5401365e-01],
[-8.5189313e-01, -2.1045120e+00, -1.4061818e+00, ...,
1.5830480e-01, 2.0167623e-01, -5.7991885e-02],
...,
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_3': layer_3input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.12081312, 0.21433187, 1.4421564 , ..., 0.48700368,
0.83675104, 2.5179884 ],
[-0.00837716, -0.60929865, 3.6444187 , ..., -1.095506 ,
-2.9168863 , 0.76664406],
[ 1.6383092 , -3.41747 , 0.1877312 , ..., -2.2291133 ,
-0.70872074, -0.67117375],
...,
[ 0.22261605, -1.6185126 , -0.29067388, ..., -0.30721062,
-0.936053 , -6.319776 ],
[ 0.96031153, 0.3580432 , -2.3184853 , ..., -3.5137513 ,
0.7674257 , 1.8014185 ],
[-1.5907912 , -1.7231196 , 0.751697 , ..., -0.34286314,
-3.2077777 , -3.3623428 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-2.7112663 , -0.9643581 , 2.9698734 , ..., -1.6670817 ,
-0.6212921 , 0.7902799 ],
[ 1.22522 , -2.1814368 , 0.55360013, ..., -0.83167875,
-0.0256775 , 1.8732722 ],
[ 0.49691385, 0.24650456, 1.1760592 , ..., 0.6746525 ,
1.6669806 , 2.3433416 ],
...,
[-4.331458 , 5.7505627 , 2.1066272 , ..., 1.3505996 ,
-2.1723619 , 2.1864228 ],
[-2.0217304 , -3.1857903 , 2.0970798 , ..., -1.8917772 ,
-1.419903 , -2.8695233 ],
[ 0.63794476, 1.5040748 , -0.4937292 , ..., 1.2936034 ,
3.5097804 , 2.3465707 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[ 0.23311885, 2.5184479 , 0.35841894, ..., 1.2846606 ,
0.6560974 , -0.598776 ],
[ 0.7402837 , 1.833609 , 2.4995692 , ..., -0.63918394,
-2.8790867 , 1.724812 ],
[ 1.0263193 , -2.7364955 , 0.9500314 , ..., -0.22096811,
1.1111536 , 0.79498106],
...,
[ 1.2500571 , 3.3613405 , 0.46471608, ..., 1.5784024 ,
3.1365042 , -2.0092516 ],
[-0.22173691, 0.08692293, -0.47295108, ..., 0.3666285 ,
-3.3435378 , 0.50542533],
[ 0.2489541 , 0.22905599, -0.8613533 , ..., -0.14420761,
2.0106645 , 0.28263944]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 1.6241203e-01 -1.1072351e+00 -2.8557816e-01 ... 3.0483809e-01
-2.0145495e-01 1.2422634e-01]
[ 1.6238174e-01 -1.1072453e+00 -2.8561494e-01 ... -1.0851421e-01
-7.4119255e-02 3.6281443e-01]
[ 1.6239472e-01 -1.1072011e+00 -2.8559628e-01 ... 1.9364634e-01
2.7551642e-01 1.8835917e-01]
...
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 1.6241203e-01, -1.1072351e+00, -2.8557816e-01, ...,
3.0483809e-01, -2.0145495e-01, 1.2422634e-01],
[ 1.6238174e-01, -1.1072453e+00, -2.8561494e-01, ...,
-1.0851421e-01, -7.4119255e-02, 3.6281443e-01],
[ 1.6239472e-01, -1.1072011e+00, -2.8559628e-01, ...,
1.9364634e-01, 2.7551642e-01, 1.8835917e-01],
...,
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-4.27108198e-01 -3.83383721e-01 -4.64053869e-01 ... 1.14207864e-01
-2.97800273e-01 -2.52104029e-02]
[-4.27099377e-01 -3.83385211e-01 -4.64054763e-01 ... -2.54104912e-01
5.22167683e-02 1.87394261e-01]
[-4.27099198e-01 -3.83387268e-01 -4.64054018e-01 ... 1.92611620e-01
2.45382249e-01 -7.05595240e-02]
...
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 4.20389539e-45 0.00000000e+00 1.40129846e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-4.27108198e-01, -3.83383721e-01, -4.64053869e-01, ...,
1.14207864e-01, -2.97800273e-01, -2.52104029e-02],
[-4.27099377e-01, -3.83385211e-01, -4.64054763e-01, ...,
-2.54104912e-01, 5.22167683e-02, 1.87394261e-01],
[-4.27099198e-01, -3.83387268e-01, -4.64054018e-01, ...,
1.92611620e-01, 2.45382249e-01, -7.05595240e-02],
...,
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 4.20389539e-45, 0.00000000e+00, 1.40129846e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[ 1.140101 , 2.0222821 , -3.1304958 , ..., 1.1466542 ,
1.7076294 , -0.21663147],
[-3.198693 , 2.8868864 , -0.3604192 , ..., -0.7823056 ,
0.6240356 , 0.98817015],
[-4.6553483 , 0.6639529 , -1.5882826 , ..., -0.1580747 ,
0.34259212, -0.9970385 ],
...,
[-0.36834115, 2.3281004 , -0.15078871, ..., -0.99443746,
2.3609061 , -2.7193348 ],
[-2.0036862 , -2.430595 , -0.5361116 , ..., -0.35383248,
-0.13564959, -1.6325415 ],
[-5.8481436 , 2.207725 , 0.9011133 , ..., -0.6096626 ,
-0.51655275, -0.6637965 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-6.1574583 , 0.5306572 , -6.2926936 , ..., -0.02864111,
-0.40215862, 5.589474 ],
[-3.3362794 , -6.298284 , -3.7303557 , ..., 2.2301488 ,
1.0625205 , 1.2262591 ],
[-2.5761666 , 1.2296215 , -0.37371475, ..., 3.42179 ,
-0.6136429 , 5.521514 ],
...,
[-2.8534966 , 0.6100542 , 2.76881 , ..., -1.0152637 ,
-4.398335 , 3.590753 ],
[-0.06075297, 1.8847165 , 4.0440626 , ..., -0.10650878,
0.69415295, -2.9796302 ],
[ 0.44205856, 3.0723534 , 1.9417149 , ..., -3.4233217 ,
2.7222688 , 0.73819387]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.9599621 , -2.1413295 , 5.519057 , ..., 0.873332 ,
3.4918246 , 3.0262551 ],
[ 1.1177975 , 2.9307213 , -3.712386 , ..., -5.507332 ,
-2.4439375 , 3.9713829 ],
[ 0.7814051 , 2.3921692 , 0.28386188, ..., 1.8599669 ,
-0.90636504, -1.0435562 ],
...,
[ 0.94891405, 0.05201203, 1.257389 , ..., 3.6583638 ,
1.3928108 , -0.59312904],
[-2.7363954 , 0.7783992 , -0.04990867, ..., 1.9942182 ,
4.239317 , -3.9982047 ],
[-1.4600741 , -1.0381477 , 2.0245078 , ..., 2.7030635 ,
1.8954883 , -0.8860114 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-7.4828275e-02 9.0286773e-01 -2.6214507e-01 ... -6.3772492e-02
3.2012242e-01 -2.6510620e-01]
[-7.4878342e-02 9.0290326e-01 -2.6211560e-01 ... 7.5849503e-02
-2.2701915e-01 -3.4585109e-01]
[-7.4837461e-02 9.0285128e-01 -2.6216862e-01 ... -4.0451840e-01
-9.3121283e-02 -6.8717144e-02]
...
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-7.4828275e-02, 9.0286773e-01, -2.6214507e-01, ...,
-6.3772492e-02, 3.2012242e-01, -2.6510620e-01],
[-7.4878342e-02, 9.0290326e-01, -2.6211560e-01, ...,
7.5849503e-02, -2.2701915e-01, -3.4585109e-01],
[-7.4837461e-02, 9.0285128e-01, -2.6216862e-01, ...,
-4.0451840e-01, -9.3121283e-02, -6.8717144e-02],
...,
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-7.4828275e-02 9.0286773e-01 -2.6214507e-01 ... -6.3772492e-02
3.2012242e-01 -2.6510620e-01]
[-7.4878342e-02 9.0290326e-01 -2.6211560e-01 ... 7.5849503e-02
-2.2701915e-01 -3.4585109e-01]
[-7.4837461e-02 9.0285128e-01 -2.6216862e-01 ... -4.0451840e-01
-9.3121283e-02 -6.8717144e-02]
...
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-7.4828275e-02, 9.0286773e-01, -2.6214507e-01, ...,
-6.3772492e-02, 3.2012242e-01, -2.6510620e-01],
[-7.4878342e-02, 9.0290326e-01, -2.6211560e-01, ...,
7.5849503e-02, -2.2701915e-01, -3.4585109e-01],
[-7.4837461e-02, 9.0285128e-01, -2.6216862e-01, ...,
-4.0451840e-01, -9.3121283e-02, -6.8717144e-02],
...,
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 1.3343583 , -0.02900918, 0.982258 , ..., -3.8649988 ,
1.6852164 , -2.8866332 ],
[ 3.663301 , -2.4428883 , 2.2433438 , ..., -3.0956182 ,
0.4433875 , -7.820587 ],
[ 4.800418 , -1.3318237 , -0.17254679, ..., 1.5635651 ,
0.811215 , -9.068334 ],
...,
[-0.16307653, -0.05829594, 4.855029 , ..., -3.9734268 ,
-7.034069 , -5.6165075 ],
[ 0.8679783 , -0.23416385, 4.688869 , ..., 1.4813001 ,
-0.89909977, -5.801539 ],
[ 1.2308747 , -0.53154147, 1.1538248 , ..., 2.3267426 ,
0.6285008 , -0.76400554]], dtype=float32):Rendering array...
,}output: Array([[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202301e-01, -4.9090710e-01, 1.1302884e+00, ...,
-1.1169595e+00, 2.4838421e-01, -1.9697804e+00],
[-4.7395071e-01, 2.9573953e-02, 1.5118444e-01, ...,
-2.0817914e+00, -7.4348491e-01, 3.5938230e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202399e-01, -4.9090755e-01, 1.1302884e+00, ...,
-1.1169568e+00, 2.4838677e-01, -1.9697807e+00],
[-4.7394973e-01, 2.9574307e-02, 1.5118416e-01, ...,
-2.0817871e+00, -7.4348247e-01, 3.5938433e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3838963e-03, 1.2510774e-02, -2.9817784e-02, ...,
-6.3862670e-03, 5.9683014e-02, -6.7729861e-02],
[ 6.0084052e-03, 9.9889673e-03, -2.5123879e-02, ...,
-4.6563623e-03, 4.8724581e-02, -5.3767640e-02]],
[[ 2.0049126e+00, -1.1143405e+00, -1.4624956e-01, ...,
2.1875381e+00, 3.9032390e+00, -1.8773682e+00],
[ 1.6202453e-01, -4.9090767e-01, 1.1302887e+00, ...,
-1.1169556e+00, 2.4838746e-01, -1.9697816e+00],
[-4.7395027e-01, 2.9573921e-02, 1.5118535e-01, ...,
-2.0817907e+00, -7.4348515e-01, 3.5938153e-01],
...,
[ 6.4780959e-03, 1.3793043e-02, -3.2066599e-02, ...,
-7.3198299e-03, 6.5112941e-02, -7.4816957e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
...,
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 2.0049131e+00 -1.1143405e+00 -1.4625034e-01 ... 2.1875381e+00
3.9032393e+00 -1.8773681e+00]
[ 1.6202301e-01 -4.9090710e-01 1.1302884e+00 ... -1.1169595e+00
2.4838421e-01 -1.9697804e+00]
[-4.7395071e-01 2.9573953e-02 1.5118444e-01 ... -2.0817914e+00
-7.4348491e-01 3.5938230e-01]
...
[ 6.4781094e-03 1.3792948e-02 -3.2066371e-02 ... -7.3198252e-03
6.5112464e-02 -7.4816644e-02]
[ 6.3839490e-03 1.2510829e-02 -2.9817669e-02 ... -6.3863890e-03
5.9683163e-02 -6.7730650e-02]
[ 6.0083596e-03 9.9889059e-03 -2.5123840e-02 ... -4.6563423e-03
4.8724260e-02 -5.3767070e-02]]
[[ 2.0049131e+00 -1.1143405e+00 -1.4625034e-01 ... 2.1875381e+00
3.9032393e+00 -1.8773681e+00]
[ 1.6202399e-01 -4.9090755e-01 1.1302884e+00 ... -1.1169568e+00
2.4838677e-01 -1.9697807e+00]
[-4.7394973e-01 2.9574307e-02 1.5118416e-01 ... -2.0817871e+00
-7.4348247e-01 3.5938433e-01]
...
[ 6.4781094e-03 1.3792948e-02 -3.2066371e-02 ... -7.3198252e-03
6.5112464e-02 -7.4816644e-02]
[ 6.3838963e-03 1.2510774e-02 -2.9817784e-02 ... -6.3862670e-03
5.9683014e-02 -6.7729861e-02]
[ 6.0084052e-03 9.9889673e-03 -2.5123879e-02 ... -4.6563623e-03
4.8724581e-02 -5.3767640e-02]]
[[ 2.0049126e+00 -1.1143405e+00 -1.4624956e-01 ... 2.1875381e+00
3.9032390e+00 -1.8773682e+00]
[ 1.6202453e-01 -4.9090767e-01 1.1302887e+00 ... -1.1169556e+00
2.4838746e-01 -1.9697816e+00]
[-4.7395027e-01 2.9573921e-02 1.5118535e-01 ... -2.0817907e+00
-7.4348515e-01 3.5938153e-01]
...
[ 6.4780959e-03 1.3793043e-02 -3.2066599e-02 ... -7.3198299e-03
6.5112941e-02 -7.4816957e-02]
[ 6.3839490e-03 1.2510829e-02 -2.9817669e-02 ... -6.3863890e-03
5.9683163e-02 -6.7730650e-02]
[ 6.0083596e-03 9.9889059e-03 -2.5123840e-02 ... -4.6563423e-03
4.8724260e-02 -5.3767070e-02]]
...
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]]key=0,): Array([[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202301e-01, -4.9090710e-01, 1.1302884e+00, ...,
-1.1169595e+00, 2.4838421e-01, -1.9697804e+00],
[-4.7395071e-01, 2.9573953e-02, 1.5118444e-01, ...,
-2.0817914e+00, -7.4348491e-01, 3.5938230e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202399e-01, -4.9090755e-01, 1.1302884e+00, ...,
-1.1169568e+00, 2.4838677e-01, -1.9697807e+00],
[-4.7394973e-01, 2.9574307e-02, 1.5118416e-01, ...,
-2.0817871e+00, -7.4348247e-01, 3.5938433e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3838963e-03, 1.2510774e-02, -2.9817784e-02, ...,
-6.3862670e-03, 5.9683014e-02, -6.7729861e-02],
[ 6.0084052e-03, 9.9889673e-03, -2.5123879e-02, ...,
-4.6563623e-03, 4.8724581e-02, -5.3767640e-02]],
[[ 2.0049126e+00, -1.1143405e+00, -1.4624956e-01, ...,
2.1875381e+00, 3.9032390e+00, -1.8773682e+00],
[ 1.6202453e-01, -4.9090767e-01, 1.1302887e+00, ...,
-1.1169556e+00, 2.4838746e-01, -1.9697816e+00],
[-4.7395027e-01, 2.9573921e-02, 1.5118535e-01, ...,
-2.0817907e+00, -7.4348515e-01, 3.5938153e-01],
...,
[ 6.4780959e-03, 1.3793043e-02, -3.2066599e-02, ...,
-7.3198299e-03, 6.5112941e-02, -7.4816957e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
...,
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[ 0.32777715 0.29137796 0.14651126 ... 0.05004397 0.16633444
-0.00615666]
[ 0.32778108 0.29137513 0.14651923 ... 0.11465082 -0.12330382
-0.04323881]
[ 0.32777962 0.2913832 0.14651957 ... -0.13493593 -0.04286611
0.10070267]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.32777715, 0.29137796, 0.14651126, ..., 0.05004397,
0.16633444, -0.00615666],
[ 0.32778108, 0.29137513, 0.14651923, ..., 0.11465082,
-0.12330382, -0.04323881],
[ 0.32777962, 0.2913832 , 0.14651957, ..., -0.13493593,
-0.04286611, 0.10070267],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[ 0.32777715 0.29137796 0.14651126 ... 0.05004397 0.16633444
-0.00615666]
[ 0.32778108 0.29137513 0.14651923 ... 0.11465082 -0.12330382
-0.04323881]
[ 0.32777962 0.2913832 0.14651957 ... -0.13493593 -0.04286611
0.10070267]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.32777715, 0.29137796, 0.14651126, ..., 0.05004397,
0.16633444, -0.00615666],
[ 0.32778108, 0.29137513, 0.14651923, ..., 0.11465082,
-0.12330382, -0.04323881],
[ 0.32777962, 0.2913832 , 0.14651957, ..., -0.13493593,
-0.04286611, 0.10070267],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 1.9669358 , 2.9423544 , 0.19496085, ..., 0.73976904,
1.6372558 , -1.2503364 ],
[-0.6908223 , 0.7870249 , 1.1638534 , ..., -0.4025112 ,
-4.771617 , 1.1944067 ],
[-0.46339673, -2.2016704 , 0.06319727, ..., 2.756306 ,
0.15175773, 5.9208183 ],
...,
[-0.37803346, 4.1615434 , -1.525748 , ..., 2.7360542 ,
0.89899254, -2.888936 ],
[ 0.03935861, 2.492835 , -0.11176654, ..., -0.69757986,
1.6231072 , -1.5887352 ],
[-0.8326037 , -2.4217205 , -2.889763 , ..., 0.8011354 ,
-1.376752 , -0.8938622 ]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[ 2.6754918 , 0.6090251 , -1.9812808 , ..., -4.481344 ,
-1.8871554 , 3.6247041 ],
[-1.1990739 , -0.38316777, 4.051298 , ..., -1.3397157 ,
-0.9165211 , -1.2980297 ],
[-0.05759763, 1.2653302 , -1.5408715 , ..., -0.81410015,
2.3423615 , 4.3030977 ],
...,
[-0.85361755, -0.14409246, 1.0777223 , ..., 1.1357626 ,
0.44438794, 2.4271173 ],
[ 0.24053141, -0.3990412 , 1.3709172 , ..., 0.8270215 ,
-0.07329489, 1.295343 ],
[-0.6063715 , -0.79813457, -0.47160372, ..., -0.5156604 ,
-0.51939523, -1.8995527 ]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[ 6.3622594 , -1.29415 , 1.2047727 , ..., 4.7288013 ,
-0.88141096, 0.18366015],
[-2.0154412 , -4.84606 , 6.55945 , ..., -4.882003 ,
2.7193513 , 6.974478 ],
[-0.93180823, 4.2919602 , 1.4147624 , ..., -0.3700378 ,
2.1703444 , 0.28357494],
...,
[-2.133009 , 4.717238 , 0.60007834, ..., 3.8875234 ,
3.0682204 , -2.3200848 ],
[-2.0814712 , 2.6006532 , -2.6173725 , ..., -5.443099 ,
2.998717 , 0.1135042 ],
[-1.8384115 , 0.4790442 , 3.2373917 , ..., -6.6073313 ,
3.9630156 , -1.5991372 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-0.06903448 -1.0456228 -0.79247475 ... -1.1273713 -0.2949175
0.00389162]
[-0.06902509 -1.045738 -0.7923472 ... -0.07797773 0.99452454
-0.601801 ]
[-0.06893523 -1.0457147 -0.7924485 ... 0.36462477 -0.6990695
-0.8575157 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.06903448, -1.0456228 , -0.79247475, ..., -1.1273713 ,
-0.2949175 , 0.00389162],
[-0.06902509, -1.045738 , -0.7923472 , ..., -0.07797773,
0.99452454, -0.601801 ],
[-0.06893523, -1.0457147 , -0.7924485 , ..., 0.36462477,
-0.6990695 , -0.8575157 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-0.06903448 -1.0456228 -0.79247475 ... -1.1273713 -0.2949175
0.00389162]
[-0.06902509 -1.045738 -0.7923472 ... -0.07797773 0.99452454
-0.601801 ]
[-0.06893523 -1.0457147 -0.7924485 ... 0.36462477 -0.6990695
-0.8575157 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.06903448, -1.0456228 , -0.79247475, ..., -1.1273713 ,
-0.2949175 , 0.00389162],
[-0.06902509, -1.045738 , -0.7923472 , ..., -0.07797773,
0.99452454, -0.601801 ],
[-0.06893523, -1.0457147 , -0.7924485 , ..., 0.36462477,
-0.6990695 , -0.8575157 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[ 9.3377553e-02 -2.1528578e+00 -1.0780529e+00 ... -8.2253325e-01
-4.9637246e-01 1.2811796e-01]
[ 9.3356647e-02 -2.1529832e+00 -1.0779622e+00 ... -1.8649194e-01
9.2040527e-01 -2.3898655e-01]
[ 9.3459487e-02 -2.1529160e+00 -1.0780448e+00 ... 5.5827111e-01
-4.2355308e-01 -6.6915655e-01]
...
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 9.3377553e-02, -2.1528578e+00, -1.0780529e+00, ...,
-8.2253325e-01, -4.9637246e-01, 1.2811796e-01],
[ 9.3356647e-02, -2.1529832e+00, -1.0779622e+00, ...,
-1.8649194e-01, 9.2040527e-01, -2.3898655e-01],
[ 9.3459487e-02, -2.1529160e+00, -1.0780448e+00, ...,
5.5827111e-01, -4.2355308e-01, -6.6915655e-01],
...,
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[ 9.3377553e-02 -2.1528578e+00 -1.0780529e+00 ... -8.2253325e-01
-4.9637246e-01 1.2811796e-01]
[ 9.3356647e-02 -2.1529832e+00 -1.0779622e+00 ... -1.8649194e-01
9.2040527e-01 -2.3898655e-01]
[ 9.3459487e-02 -2.1529160e+00 -1.0780448e+00 ... 5.5827111e-01
-4.2355308e-01 -6.6915655e-01]
...
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 9.3377553e-02, -2.1528578e+00, -1.0780529e+00, ...,
-8.2253325e-01, -4.9637246e-01, 1.2811796e-01],
[ 9.3356647e-02, -2.1529832e+00, -1.0779622e+00, ...,
-1.8649194e-01, 9.2040527e-01, -2.3898655e-01],
[ 9.3459487e-02, -2.1529160e+00, -1.0780448e+00, ...,
5.5827111e-01, -4.2355308e-01, -6.6915655e-01],
...,
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_4': layer_4input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 16384 parametersw[0,0] 128x0e,128x0e: Array([[ 0.3937284 , 0.06909973, 0.18853222, ..., -0.10455237,
-0.071788 , 0.19546314],
[ 0.61504227, -0.43697816, 0.1261993 , ..., 0.35577676,
0.9733432 , 0.18279058],
[-0.6915464 , 1.3660764 , 0.5243588 , ..., 0.05164947,
0.03794032, 1.1182275 ],
...,
[-0.6308902 , -1.835251 , -0.13734646, ..., 0.76688886,
0.05780329, -0.44389218],
[ 0.06009359, -3.1872697 , 0.38784972, ..., 0.22112429,
-0.03782988, -0.42557165],
[-0.5971681 , -0.8687849 , -0.3203398 , ..., 0.63645226,
-0.05153882, -0.44785336]], dtype=float32):Rendering array...
,}output: 128x0e
[[ 1.8583839e-01 1.8836857e-01 3.1776994e-01 ... -3.5417807e-01
1.6731169e-02 -2.9344723e-01]
[ 1.8584874e-01 1.8838334e-01 3.1776556e-01 ... -3.5415211e-01
1.6726103e-02 -2.9343879e-01]
[ 1.8581046e-01 1.8845277e-01 3.1775710e-01 ... -3.5414886e-01
1.6718570e-02 -2.9342994e-01]
...
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]]key=0,): Array([[ 1.8583839e-01, 1.8836857e-01, 3.1776994e-01, ...,
-3.5417807e-01, 1.6731169e-02, -2.9344723e-01],
[ 1.8584874e-01, 1.8838334e-01, 3.1776556e-01, ...,
-3.5415211e-01, 1.6726103e-02, -2.9343879e-01],
[ 1.8581046e-01, 1.8845277e-01, 3.1775710e-01, ...,
-3.5414886e-01, 1.6718570e-02, -2.9342994e-01],
...,
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44]], dtype=float32), :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[ 8.1928745e-02 -3.7561089e-01 -4.5885956e-01 ... 2.6948872e-01
1.6262782e-01 -4.1975625e-02]
[ 8.1909589e-02 -3.7559056e-01 -4.5885199e-01 ... 6.1089247e-02
-3.0149758e-01 7.8284934e-02]
[ 8.2003839e-02 -3.7560147e-01 -4.5885891e-01 ... -1.8289228e-01
1.3875800e-01 2.1921887e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 8.1928745e-02, -3.7561089e-01, -4.5885956e-01, ...,
2.6948872e-01, 1.6262782e-01, -4.1975625e-02],
[ 8.1909589e-02, -3.7559056e-01, -4.5885199e-01, ...,
6.1089247e-02, -3.0149758e-01, 7.8284934e-02],
[ 8.2003839e-02, -3.7560147e-01, -4.5885891e-01, ...,
-1.8289228e-01, 1.3875800e-01, 2.1921887e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[ 2.7310824 , 3.0626702 , -2.873125 , ..., -0.46205166,
-2.5823078 , -0.79186493],
[ 1.5819222 , -0.04187162, 0.18020032, ..., -1.421179 ,
0.36783072, -0.41777498],
[ 2.1611493 , -3.5538318 , -3.4319103 , ..., -4.8217306 ,
9.292773 , 1.6516443 ],
...,
[-0.2557063 , -0.08857636, -1.337759 , ..., -0.20876619,
-1.4309722 , -1.1555536 ],
[-0.39878693, 2.476265 , 0.19616641, ..., -0.9218898 ,
1.2366259 , -0.43310922],
[ 0.22334357, 2.3639684 , 1.1262287 , ..., -0.26591426,
0.29230833, -1.8359554 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 2.957393 , -1.2974124 , -1.8927755 , ..., -1.4519069 ,
0.89650595, 2.8739493 ],
[ 2.376374 , -0.32495666, -0.89615333, ..., -0.30789825,
-0.12491573, 1.2192786 ],
[ 1.0807056 , -0.37785774, 1.0011708 , ..., 0.84363145,
0.42042476, -0.62347543],
...,
[-3.6531384 , 1.5399444 , -2.8130946 , ..., 3.2157907 ,
0.9592581 , -1.5591872 ],
[ 3.7486699 , 2.8656209 , -0.21817568, ..., 0.48192185,
0.4373116 , 2.5665076 ],
[ 2.39491 , 2.5794294 , 1.1376072 , ..., -2.5818353 ,
-5.080429 , -0.04517885]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[ 1.834679 , -3.5193052 , 0.28963464, ..., 0.7220885 ,
-1.037215 , 1.7926254 ],
[-2.8789775 , 3.5531743 , -1.5598752 , ..., 0.52459097,
1.1114972 , 5.42312 ],
[ 0.54869986, 0.5619909 , -4.4183044 , ..., 0.17171204,
3.3009405 , 0.10807901],
...,
[-2.351306 , -0.04581549, -0.554534 , ..., 2.7257571 ,
0.9586024 , 4.887769 ],
[ 0.21695383, 1.2257326 , -1.1554599 , ..., -0.86416173,
-1.4262117 , 1.0316494 ],
[-3.827227 , 1.203004 , 0.61138934, ..., 0.2602603 ,
-4.1413083 , -3.2767403 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-8.39686573e-01 2.13831607e-02 -1.02276362e-01 ... 2.29670000e+00
1.29684997e+00 -8.97384167e-01]
[-8.39542091e-01 2.13457979e-02 -1.02192916e-01 ... 1.03079602e-01
-2.76034379e+00 3.69943917e-01]
[-8.39598954e-01 2.14124955e-02 -1.02234721e-01 ... -1.84424925e+00
1.46188486e+00 1.49303889e+00]
...
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-8.39686573e-01, 2.13831607e-02, -1.02276362e-01, ...,
2.29670000e+00, 1.29684997e+00, -8.97384167e-01],
[-8.39542091e-01, 2.13457979e-02, -1.02192916e-01, ...,
1.03079602e-01, -2.76034379e+00, 3.69943917e-01],
[-8.39598954e-01, 2.14124955e-02, -1.02234721e-01, ...,
-1.84424925e+00, 1.46188486e+00, 1.49303889e+00],
...,
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-8.39686573e-01 2.13831607e-02 -1.02276362e-01 ... 2.29670000e+00
1.29684997e+00 -8.97384167e-01]
[-8.39542091e-01 2.13457979e-02 -1.02192916e-01 ... 1.03079602e-01
-2.76034379e+00 3.69943917e-01]
[-8.39598954e-01 2.14124955e-02 -1.02234721e-01 ... -1.84424925e+00
1.46188486e+00 1.49303889e+00]
...
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-1.54142831e-44 -6.02558340e-44 1.54142831e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-8.39686573e-01, 2.13831607e-02, -1.02276362e-01, ...,
2.29670000e+00, 1.29684997e+00, -8.97384167e-01],
[-8.39542091e-01, 2.13457979e-02, -1.02192916e-01, ...,
1.03079602e-01, -2.76034379e+00, 3.69943917e-01],
[-8.39598954e-01, 2.14124955e-02, -1.02234721e-01, ...,
-1.84424925e+00, 1.46188486e+00, 1.49303889e+00],
...,
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-1.54142831e-44, -6.02558340e-44, 1.54142831e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 1792 parameterskernel: Array([[ 0.26945972, 0.43817306, 0.03105241, ..., 0.19044442,
0.2351014 , 0.03229111],
[ 0.7108974 , 0.93321556, 0.3339254 , ..., -0.10694128,
0.47263741, 0.99574834],
[ 0.35409585, 1.0157957 , 0.28180015, ..., 0.22329491,
-0.40699825, 1.0774257 ],
...,
[ 0.32090774, -0.531131 , 0.1540587 , ..., -0.4244127 ,
0.31799552, 0.2742479 ],
[ 0.29262787, -0.6745612 , -0.04605831, ..., -0.00225773,
0.02264951, 0.49221578],
[ 0.0800835 , -0.36611298, 0.12036715, ..., 0.13340048,
-0.16945608, 0.12831876]], dtype=float32):Rendering array...
,}output: Array([[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239378e-02, 1.99974403e-01, -4.91603874e-02, ...,
-1.46051705e-01, 6.90582171e-02, 2.65027583e-01],
[ 4.36086990e-02, -4.75874059e-02, 2.28474177e-02, ...,
-3.79332565e-02, 1.89851657e-01, -2.00742885e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239304e-02, 1.99974477e-01, -4.91604134e-02, ...,
-1.46051615e-01, 6.90580159e-02, 2.65027851e-01],
[ 4.36086841e-02, -4.75880094e-02, 2.28475034e-02, ...,
-3.79327834e-02, 1.89851508e-01, -2.00743139e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124566e-03, 4.81318729e-03, -5.20544825e-03, ...,
7.13166595e-03, -1.04736676e-02, -3.22475052e-03],
[-3.44700669e-03, 3.99699714e-03, -4.09354130e-03, ...,
5.87636186e-03, -8.71226937e-03, -2.69638211e-03]],
[[ 1.04371056e-01, 5.70607960e-01, 4.65185242e-03, ...,
9.47357640e-02, -2.08477169e-01, 5.68402112e-01],
[ 9.81239602e-02, 1.99974552e-01, -4.91604246e-02, ...,
-1.46051586e-01, 6.90579042e-02, 2.65028059e-01],
[ 4.36087660e-02, -4.75876220e-02, 2.28474680e-02, ...,
-3.79331931e-02, 1.89851642e-01, -2.00742856e-01],
...,
[-4.51123063e-03, 5.21379616e-03, -5.77482814e-03, ...,
7.74761708e-03, -1.13302628e-02, -3.48037179e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
...,
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]]], dtype=float32):Rendering array...
,),}output: 224x0e
[[[ 1.04371041e-01 5.70608139e-01 4.65189153e-03 ... 9.47358012e-02
-2.08477199e-01 5.68402112e-01]
[ 9.81239378e-02 1.99974403e-01 -4.91603874e-02 ... -1.46051705e-01
6.90582171e-02 2.65027583e-01]
[ 4.36086990e-02 -4.75874059e-02 2.28474177e-02 ... -3.79332565e-02
1.89851657e-01 -2.00742885e-01]
...
[-4.51120175e-03 5.21376915e-03 -5.77478809e-03 ... 7.74756679e-03
-1.13302087e-02 -3.48034035e-03]
[-4.16124752e-03 4.81318217e-03 -5.20547898e-03 ... 7.13169714e-03
-1.04737105e-02 -3.22474353e-03]
[-3.44699319e-03 3.99698783e-03 -4.09350777e-03 ... 5.87632274e-03
-8.71221721e-03 -2.69637932e-03]]
[[ 1.04371041e-01 5.70608139e-01 4.65189153e-03 ... 9.47358012e-02
-2.08477199e-01 5.68402112e-01]
[ 9.81239304e-02 1.99974477e-01 -4.91604134e-02 ... -1.46051615e-01
6.90580159e-02 2.65027851e-01]
[ 4.36086841e-02 -4.75880094e-02 2.28475034e-02 ... -3.79327834e-02
1.89851508e-01 -2.00743139e-01]
...
[-4.51120175e-03 5.21376915e-03 -5.77478809e-03 ... 7.74756679e-03
-1.13302087e-02 -3.48034035e-03]
[-4.16124566e-03 4.81318729e-03 -5.20544825e-03 ... 7.13166595e-03
-1.04736676e-02 -3.22475052e-03]
[-3.44700669e-03 3.99699714e-03 -4.09354130e-03 ... 5.87636186e-03
-8.71226937e-03 -2.69638211e-03]]
[[ 1.04371056e-01 5.70607960e-01 4.65185242e-03 ... 9.47357640e-02
-2.08477169e-01 5.68402112e-01]
[ 9.81239602e-02 1.99974552e-01 -4.91604246e-02 ... -1.46051586e-01
6.90579042e-02 2.65028059e-01]
[ 4.36087660e-02 -4.75876220e-02 2.28474680e-02 ... -3.79331931e-02
1.89851642e-01 -2.00742856e-01]
...
[-4.51123063e-03 5.21379616e-03 -5.77482814e-03 ... 7.74761708e-03
-1.13302628e-02 -3.48037179e-03]
[-4.16124752e-03 4.81318217e-03 -5.20547898e-03 ... 7.13169714e-03
-1.04737105e-02 -3.22474353e-03]
[-3.44699319e-03 3.99698783e-03 -4.09350777e-03 ... 5.87632274e-03
-8.71221721e-03 -2.69637932e-03]]
...
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]]key=0,): Array([[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239378e-02, 1.99974403e-01, -4.91603874e-02, ...,
-1.46051705e-01, 6.90582171e-02, 2.65027583e-01],
[ 4.36086990e-02, -4.75874059e-02, 2.28474177e-02, ...,
-3.79332565e-02, 1.89851657e-01, -2.00742885e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239304e-02, 1.99974477e-01, -4.91604134e-02, ...,
-1.46051615e-01, 6.90580159e-02, 2.65027851e-01],
[ 4.36086841e-02, -4.75880094e-02, 2.28475034e-02, ...,
-3.79327834e-02, 1.89851508e-01, -2.00743139e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124566e-03, 4.81318729e-03, -5.20544825e-03, ...,
7.13166595e-03, -1.04736676e-02, -3.22475052e-03],
[-3.44700669e-03, 3.99699714e-03, -4.09354130e-03, ...,
5.87636186e-03, -8.71226937e-03, -2.69638211e-03]],
[[ 1.04371056e-01, 5.70607960e-01, 4.65185242e-03, ...,
9.47357640e-02, -2.08477169e-01, 5.68402112e-01],
[ 9.81239602e-02, 1.99974552e-01, -4.91604246e-02, ...,
-1.46051586e-01, 6.90579042e-02, 2.65028059e-01],
[ 4.36087660e-02, -4.75876220e-02, 2.28474680e-02, ...,
-3.79331931e-02, 1.89851642e-01, -2.00742856e-01],
...,
[-4.51123063e-03, 5.21379616e-03, -5.77482814e-03, ...,
7.74761708e-03, -1.13302628e-02, -3.48037179e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
...,
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]]], dtype=float32), :Rendering array...
,,),}output: 224x0e
[[ 0.00020877 0.08437879 0.00556198 ... 0.012841 0.03463268
-0.09035657]
[ 0.00020891 0.08438315 0.00556233 ... 0.01283049 0.03469033
-0.09047941]
[ 0.00020593 0.08437952 0.00556294 ... 0.01283367 0.03465934
-0.09045679]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00020877, 0.08437879, 0.00556198, ..., 0.012841 ,
0.03463268, -0.09035657],
[ 0.00020891, 0.08438315, 0.00556233, ..., 0.01283049,
0.03469033, -0.09047941],
[ 0.00020593, 0.08437952, 0.00556294, ..., 0.01283367,
0.03465934, -0.09045679],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e
[[ 0.00020877 0.08437879 0.00556198 ... 0.012841 0.03463268
-0.09035657]
[ 0.00020891 0.08438315 0.00556233 ... 0.01283049 0.03469033
-0.09047941]
[ 0.00020593 0.08437952 0.00556294 ... 0.01283367 0.03465934
-0.09045679]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00020877, 0.08437879, 0.00556198, ..., 0.012841 ,
0.03463268, -0.09035657],
[ 0.00020891, 0.08438315, 0.00556233, ..., 0.01283049,
0.03469033, -0.09047941],
[ 0.00020593, 0.08437952, 0.00556294, ..., 0.01283367,
0.03465934, -0.09045679],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,}params: 28672 parametersw[0,0] 224x0e,128x0e: Array([[-1.0418724 , -0.41554564, -0.68577147, ..., -0.3166389 ,
0.8383788 , 0.1606412 ],
[-1.3690069 , 1.2751813 , 0.5149619 , ..., 3.0915513 ,
0.19957714, -2.5659294 ],
[ 0.9293653 , 1.6508423 , 0.77863246, ..., 0.11865105,
0.87970626, -0.9510862 ],
...,
[ 1.1623504 , -0.6409943 , -0.52440107, ..., 0.7576407 ,
-1.0330133 , -1.2527919 ],
[ 0.48919797, 1.302377 , -0.38269427, ..., 0.03654883,
0.9190065 , 1.2383986 ],
[-0.7033828 , -0.35301045, -0.9827658 , ..., 0.5619353 ,
-0.15537892, 0.10997062]], dtype=float32):Rendering array...
,}output: 128x0e
[[-0.0829545 0.18034497 -0.11269625 ... -0.19836178 0.07025553
-0.05102922]
[-0.08299188 0.18032666 -0.1127164 ... -0.19833194 0.07025995
-0.05091575]
[-0.08296126 0.18041018 -0.11268901 ... -0.19834885 0.07027565
-0.05098148]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.0829545 , 0.18034497, -0.11269625, ..., -0.19836178,
0.07025553, -0.05102922],
[-0.08299188, 0.18032666, -0.1127164 , ..., -0.19833194,
0.07025995, -0.05091575],
[-0.08296126, 0.18041018, -0.11268901, ..., -0.19834885,
0.07027565, -0.05098148],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[-0.0829545 0.18034497 -0.11269625 ... -0.19836178 0.07025553
-0.05102922]
[-0.08299188 0.18032666 -0.1127164 ... -0.19833194 0.07025995
-0.05091575]
[-0.08296126 0.18041018 -0.11268901 ... -0.19834885 0.07027565
-0.05098148]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.0829545 , 0.18034497, -0.11269625, ..., -0.19836178,
0.07025553, -0.05102922],
[-0.08299188, 0.18032666, -0.1127164 , ..., -0.19833194,
0.07025995, -0.05091575],
[-0.08296126, 0.18041018, -0.11268901, ..., -0.19834885,
0.07027565, -0.05098148],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[ 1.0288389e-01 3.6871356e-01 2.0507370e-01 ... -5.5253983e-01
8.6986691e-02 -3.4447646e-01]
[ 1.0285686e-01 3.6870998e-01 2.0504916e-01 ... -5.5248404e-01
8.6986050e-02 -3.4435454e-01]
[ 1.0284920e-01 3.6886293e-01 2.0506808e-01 ... -5.5249774e-01
8.6994216e-02 -3.4441143e-01]
...
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]]key=0,): Array([[ 1.0288389e-01, 3.6871356e-01, 2.0507370e-01, ...,
-5.5253983e-01, 8.6986691e-02, -3.4447646e-01],
[ 1.0285686e-01, 3.6870998e-01, 2.0504916e-01, ...,
-5.5248404e-01, 8.6986050e-02, -3.4435454e-01],
[ 1.0284920e-01, 3.6886293e-01, 2.0506808e-01, ...,
-5.5249774e-01, 8.6994216e-02, -3.4441143e-01],
...,
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44]], dtype=float32), :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 128x0e
[[ 1.0288389e-01 3.6871356e-01 2.0507370e-01 ... -5.5253983e-01
8.6986691e-02 -3.4447646e-01]
[ 1.0285686e-01 3.6870998e-01 2.0504916e-01 ... -5.5248404e-01
8.6986050e-02 -3.4435454e-01]
[ 1.0284920e-01 3.6886293e-01 2.0506808e-01 ... -5.5249774e-01
8.6994216e-02 -3.4441143e-01]
...
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]]key=0,): Array([[ 1.0288389e-01, 3.6871356e-01, 2.0507370e-01, ...,
-5.5253983e-01, 8.6986691e-02, -3.4447646e-01],
[ 1.0285686e-01, 3.6870998e-01, 2.0504916e-01, ...,
-5.5248404e-01, 8.6986050e-02, -3.4435454e-01],
[ 1.0284920e-01, 3.6886293e-01, 2.0506808e-01, ...,
-5.5249774e-01, 8.6994216e-02, -3.4441143e-01],
...,
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44]], dtype=float32), :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e
[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]
[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]
[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,), 'readout': readoutinput: 'x': 128x0e
[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]
[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]
[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,}output: 128x0e
[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]
[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]
[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]
[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]
[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,, 128x0e
[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]
[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]
[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,),),}output: 128x0e
[[[ 9.0678498e-02 3.6543128e-01 1.8947524e-01 ... -3.3837971e-01
7.6088928e-02 -2.3951761e-01]]
[[ 9.0653509e-02 3.6542720e-01 1.8945047e-01 ... -3.3835751e-01
7.6088339e-02 -2.3944996e-01]]
[[ 9.0646431e-02 3.6560166e-01 1.8946956e-01 ... -3.3836296e-01
7.6095775e-02 -2.3948154e-01]]
...
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]]key=0,): Array([[[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01]],
[[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01]],
[[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01]],
...,
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]],
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]],
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]]], dtype=float32), :Rendering array...
,,), 'head_dropout': head_dropoutinput: 'inputs': Array([[ 9.0678498e-02, 3.6543128e-01, 1.8947524e-01, ...,
-3.3837971e-01, 7.6088928e-02, -2.3951761e-01],
[ 9.0653509e-02, 3.6542720e-01, 1.8945047e-01, ...,
-3.3835751e-01, 7.6088339e-02, -2.3944996e-01],
[ 9.0646431e-02, 3.6560166e-01, 1.8946956e-01, ...,
-3.3836296e-01, 7.6095775e-02, -2.3948154e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32):Rendering array...
, 'deterministic': False,}output: Array([[ 1.0075389e-01, 4.0603477e-01, 0.0000000e+00, ...,
-3.7597749e-01, 8.4543258e-02, -2.6613069e-01],
[ 1.0072613e-01, 4.0603024e-01, 2.1050054e-01, ...,
-3.7595281e-01, 8.4542602e-02, -2.6605552e-01],
[ 1.0071826e-01, 4.0622407e-01, 2.1052174e-01, ...,
-3.7595886e-01, 8.4550865e-02, -2.6609060e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,), 'head': headinput: 'x': Array([[ 1.0075389e-01, 4.0603477e-01, 0.0000000e+00, ...,
-3.7597749e-01, 8.4543258e-02, -2.6613069e-01],
[ 1.0072613e-01, 4.0603024e-01, 2.1050054e-01, ...,
-3.7595281e-01, 8.4542602e-02, -2.6605552e-01],
[ 1.0071826e-01, 4.0622407e-01, 2.1052174e-01, ...,
-3.7595886e-01, 8.4550865e-02, -2.6609060e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,}children: 'Dense_0': Dense_0input: 'inputs': Array([[ 1.0075389e-01, 4.0603477e-01, 0.0000000e+00, ...,
-3.7597749e-01, 8.4543258e-02, -2.6613069e-01],
[ 1.0072613e-01, 4.0603024e-01, 2.1050054e-01, ...,
-3.7595281e-01, 8.4542602e-02, -2.6605552e-01],
[ 1.0071826e-01, 4.0622407e-01, 2.1052174e-01, ...,
-3.7595886e-01, 8.4550865e-02, -2.6609060e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,}params: 128 parameterskernel: Array([[-7.3323093e-02],
[ 2.0857520e-02],
[-7.4718341e-02],
[-6.2562965e-02],
[ 4.9506143e-02],
[ 6.4088538e-02],
[-2.1354200e-02],
[ 4.1273855e-02],
[-8.7901473e-02],
[-7.6430246e-02],
[-5.6931920e-02],
[-7.9819717e-02],
[ 4.6707336e-02],
[-1.0819524e-03],
[-4.9101446e-02],
[-4.7692545e-03],
[-6.8938918e-02],
[ 4.4658501e-02],
[ 5.2118856e-02],
[ 5.2286707e-02],
[-2.1002230e-03],
[-1.1098199e-03],
[ 6.4877555e-02],
[-5.3715833e-02],
[ 5.3442709e-02],
[-1.5025394e-03],
[ 5.8496479e-02],
[-6.8124473e-02],
[-7.5792834e-02],
[-7.0392489e-02],
[ 2.5195791e-03],
[-6.4322315e-02],
[-8.0225900e-02],
[ 5.1826373e-02],
[-6.8810103e-03],
[ 5.3192981e-02],
[-7.2647415e-02],
[-5.9058398e-02],
[-6.0489949e-02],
[ 5.6477930e-02],
[ 2.9234491e-02],
[-7.9039140e-03],
[ 1.3872646e-04],
[-6.3473918e-02],
[ 5.2041385e-02],
[ 4.2663690e-02],
[-6.8757080e-02],
[-7.6088086e-02],
[-4.5156661e-02],
[-3.1352520e-02],
[-7.5085357e-02],
[ 4.7567796e-02],
[-6.4098891e-03],
[ 2.0447827e-03],
[-6.5735050e-02],
[ 5.3788118e-02],
[-3.9345171e-02],
[-5.1456850e-02],
[ 5.3774357e-02],
[ 2.8405490e-03],
[-5.4608274e-05],
[-8.2065895e-02],
[-8.2310170e-02],
[-5.8823891e-02],
[ 4.0307153e-02],
[ 4.6815358e-02],
[-6.5950818e-02],
[ 5.0985575e-02],
[ 5.9336908e-02],
[-2.6204260e-03],
[-5.7491747e-04],
[-6.8728998e-02],
[-7.5307488e-02],
[ 4.6691045e-02],
[-8.5048757e-02],
[-7.2264649e-02],
[-2.3038473e-02],
[-7.1395472e-02],
[ 3.2673776e-03],
[ 4.2771209e-02],
[ 5.5624627e-02],
[-2.1413467e-03],
[-4.0388420e-02],
[ 1.2444846e-03],
[-5.2980776e-04],
[ 2.5609902e-03],
[ 3.7861802e-02],
[ 5.5019349e-02],
[-2.2600594e-03],
[-5.9719913e-02],
[-5.6984358e-02],
[ 4.7067355e-02],
[-2.2469580e-03],
[-6.7480549e-02],
[-9.1918424e-02],
[-5.4613639e-02],
[ 4.6480380e-02],
[-6.2892988e-02],
[ 5.2705746e-02],
[-7.3074378e-02],
[ 4.4752839e-03],
[ 5.2366756e-02],
[-3.6038421e-02],
[ 6.0057804e-02],
[ 4.4242196e-02],
[ 4.7954105e-02],
[-1.8862605e-03],
[ 5.4067023e-02],
[-7.5417444e-02],
[-3.2411609e-03],
[-7.7518038e-02],
[-8.6692937e-02],
[-5.1371388e-02],
[-3.8555808e-02],
[ 3.9256103e-02],
[ 6.9600618e-03],
[ 4.0010110e-02],
[-3.0875580e-02],
[-6.9769055e-02],
[-1.2040902e-03],
[-6.6607162e-02],
[-2.6735226e-03],
[ 4.8590209e-02],
[ 4.9908720e-02],
[ 4.9790747e-02],
[ 5.8867864e-02],
[-7.4486784e-04],
[ 3.8053673e-02]], dtype=float32):Rendering array...
,}output: Array([[-1.3227775e+00],
[-1.3328015e+00],
[-1.3621812e+00],
...,
[-5.4650640e-44],
[-5.6051939e-44],
[-5.0446745e-44]], dtype=float32):Rendering array...
,),}output: Array([[-1.3227775e+00],
[-1.3328015e+00],
[-1.3621812e+00],
...,
[-5.4650640e-44],
[-5.6051939e-44],
[-5.0446745e-44]], dtype=float32):Rendering array...
,), 'rescale': rescaleinput: 'cg': nodes=species=Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, cart=Array([[-1.6074809 , 3.990498 , 0.59335387],
[13.654599 , 0.6031299 , 4.7416563 ],
[ 2.6521325 , 3.3873684 , 4.7416563 ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, graph_i=Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
,), edges=to_jimage=Array([[[-1, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, -1]],
[[ 0, -1, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 1, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
...,
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]]], dtype=int8):Rendering array...
, receiver=Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,), n_node=Array([168, 80, 68, 60, 53, 48, 42, 40, 36, 32, 31, 28, 26,
24, 24, 21, 20, 18, 16, 16, 14, 12, 12, 10, 9, 8,
6, 5, 4, 4, 1, 88], dtype=uint16):Rendering array...
, padding_mask=Array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, False], dtype=bool):Rendering array...
, graph_data=dataset_id=Array([ 114536034, 1780206004, -1687275888, 1202003, 1758626004,
1556291002, 139460000, 1554815003, 118640006, 1675109005,
-1651451869, 1555003018, 1615850000, -1709938884, -1592934888,
-1650297867, -1696810885, -1663817880, -1696683861, -1710558886,
-1662911874, -1781222858, 123675020, -1806109888, -1804816886,
-1875814888, -1668181878, 122133000, 1864969002, -1700082886,
120483000, 0], dtype=int32):Rendering array...
, abc=Array([[17.914213 , 17.914213 , 8.296605 ],
[10.275203 , 16.84246 , 5.0821056],
[ 6.74365 , 9.197391 , 12.196357 ],
[ 9.433409 , 9.433409 , 20.947056 ],
[10.816606 , 10.426013 , 10.417979 ],
[ 6.8768296, 11.247085 , 19.015215 ],
[ 9.827564 , 9.827564 , 6.762959 ],
[ 6.386287 , 6.386287 , 12.2509365],
[ 8.614349 , 8.614349 , 8.614349 ],
[ 7.9092207, 7.7653685, 6.9275327],
[ 8.073368 , 20.393888 , 10.60539 ],
[ 5.306318 , 9.112615 , 10.040712 ],
[ 5.627906 , 6.953792 , 10.501416 ],
[ 5.037883 , 5.8846297, 7.8679 ],
[ 5.66 , 5.650074 , 9.156885 ],
[ 6.1716037, 8.621225 , 5.767166 ],
[ 6.2029986, 8.613454 , 8.613454 ],
[ 5.494526 , 5.494526 , 7.4353304],
[ 3.7546806, 4.811737 , 5.4126735],
[ 2.715433 , 5.2583437, 9.388806 ],
[ 6.0365143, 6.034208 , 6.02813 ],
[ 5.419802 , 6.49084 , 6.660221 ],
[ 4.117953 , 5.849566 , 5.849566 ],
[ 4.299559 , 4.299559 , 10.174184 ],
[ 4.235417 , 7.5434947, 7.5435066],
[ 6.565039 , 6.565039 , 6.565039 ],
[ 4.3139734, 5.5680494, 5.484443 ],
[ 5.9383936, 5.9383936, 5.9383936],
[ 4.4320164, 4.4320164, 4.4320164],
[ 3.640625 , 3.640625 , 6.0473022],
[ 3.571267 , 3.571267 , 3.571267 ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, angles_rad=Array([[1.5707964, 1.5707964, 2.0943952],
[1.5707964, 1.5707964, 1.5707964],
[1.2343053, 1.5818529, 1.1629694],
[1.5707964, 1.5707964, 1.5707964],
[1.8930188, 1.8607063, 2.0151982],
[1.7593405, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 2.0943952],
[1.730846 , 1.730846 , 1.8141689],
[1.0478324, 1.0478324, 1.0478324],
[1.8424883, 1.5707964, 1.5707964],
[0.3288564, 1.0035257, 1.0851887],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[1.7750735, 1.7689804, 1.2668823],
[1.6448978, 1.4945464, 1.8871104],
[1.5798758, 1.5553586, 1.5304748],
[1.2806493, 1.2024467, 1.2024467],
[1.2456238, 1.397332 , 1.9114691],
[1.5707964, 1.5707964, 1.5707964],
[1.644386 , 1.5707964, 1.5707964],
[1.5746922, 2.094859 , 1.0429444],
[2.0565953, 1.3555512, 1.5138267],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[2.0943942, 1.5707964, 1.5707964],
[2.2965217, 2.2965217, 1.237292 ],
[1.6661328, 1.9750055, 1.1730014],
[2.4071865, 2.4071865, 1.0650531],
[1.0471976, 1.0471976, 1.0471976],
[1.5707964, 1.5707964, 2.0943952],
[1.0471976, 1.0471976, 1.0471976],
[0. , 0. , 0. ]], dtype=float32):Rendering array...
, lat=Array([[[ 1.7914213e+01, 1.5000001e-07, 0.0000000e+00],
[-8.9571066e+00, 1.5514163e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 8.2966051e+00]],
[[ 1.0275203e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.6842461e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.0821056e+00]],
[[ 1.6343699e-01, -1.7300400e-01, 6.7394490e+00],
[ 8.5276871e+00, 6.0145002e-02, 3.4448421e+00],
[ 4.2400112e+00, 1.1435485e+01, 5.5798002e-02]],
[[ 9.4334087e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.4334087e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 2.0947056e+01]],
[[-6.3495708e+00, 6.5125003e+00, 5.8539939e+00],
[ 6.0938725e+00, -6.4372239e+00, 5.4889545e+00],
[ 6.0836735e+00, 6.1772451e+00, -5.7762318e+00]],
[[ 6.8768296e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.1048177e+01, -2.1058772e+00],
[ 0.0000000e+00, -3.7062699e-03, 1.9015215e+01]],
[[ 4.9137821e+00, -8.5109196e+00, 0.0000000e+00],
[ 4.9137821e+00, 8.5109196e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7629590e+00]],
[[ 6.3021402e+00, -6.3853199e-03, -1.0332731e+00],
[-1.7227671e+00, 6.0621023e+00, -1.0332731e+00],
[ 3.0169221e-02, 3.9895471e-02, 1.2250834e+01]],
[[ 7.4619780e+00, -1.4151999e-03, 4.3041716e+00],
[ 2.4855030e+00, 7.0358648e+00, 4.3041716e+00],
[-2.0012499e-03, -1.4151999e-03, 8.6143484e+00]],
[[ 7.9092207e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 7.5781007e+00, -1.6950918e+00],
[ 0.0000000e+00, -3.5751951e-01, 6.9183011e+00]],
[[ 6.7416244e+00, 5.3566109e-02, 4.4414983e+00],
[-1.2468886e+00, 7.0104933e+00, 1.9110441e+01],
[-1.6173536e-01, 2.7245510e-01, 1.0600656e+01]],
[[ 5.3063178e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.1126146e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0040712e+01]],
[[ 5.6279058e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 6.9537921e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0501416e+01]],
[[ 4.9857669e+00, 3.5377501e-03, -7.2275919e-01],
[ 1.6229881e+00, 5.5577154e+00, -1.0519388e+00],
[-4.2640892e-01, -7.8503363e-02, 7.8559442e+00]],
[[-4.6255998e-02, 4.7460151e+00, 3.0836339e+00],
[ 4.7340288e+00, -4.6193000e-02, -3.0838571e+00],
[-4.7798352e+00, 4.7923450e+00, -6.1672649e+00]],
[[ 6.1701546e+00, 1.2492373e-01, 4.7747571e-02],
[ 1.7339531e-01, 8.6193953e+00, -3.8590480e-02],
[ 4.4977799e-02, -2.7458400e-02, 5.7669253e+00]],
[[ 0.0000000e+00, 9.9999999e-09, -6.2029986e+00],
[-5.1461577e+00, 6.1716561e+00, -3.1015000e+00],
[ 5.1461577e+00, 6.1716561e+00, -3.1015000e+00]],
[[-3.9119451e+00, -3.1703908e+00, 2.1988914e+00],
[ 3.9119451e+00, -3.1703908e+00, -2.1988914e+00],
[ 3.8619483e+00, -3.1703908e+00, 5.5061889e+00]],
[[ 0.0000000e+00, -3.7546806e+00, 0.0000000e+00],
[-4.8117371e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -5.4126735e+00]],
[[ 2.7154329e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.2441120e+00, -3.8661101e-01],
[ 0.0000000e+00, 0.0000000e+00, 9.3888063e+00]],
[[-4.2616282e+00, -8.0763102e-03, -4.2752743e+00],
[-8.2218703e-03, -4.2585020e+00, -4.2751322e+00],
[-3.6448999e-03, -4.2542191e+00, 4.2708263e+00]],
[[ 5.3959742e+00, 2.2922444e-01, -4.5296320e-01],
[-7.0430361e-02, -1.8796436e+00, -6.2123251e+00],
[ 1.2653049e+00, 6.4092894e+00, 1.2955921e+00]],
[[ 4.1179528e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.8495660e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.8495660e+00]],
[[ 4.2995591e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995591e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0174184e+01]],
[[ 0.0000000e+00, 0.0000000e+00, -4.2354169e+00],
[-3.7717474e+00, -6.5328579e+00, 0.0000000e+00],
[-3.7717595e+00, 6.5328650e+00, 0.0000000e+00]],
[[-2.6921539e+00, 2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, -2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, 2.6921539e+00, -5.3483038e+00]],
[[-4.3139734e+00, 0.0000000e+00, 0.0000000e+00],
[-2.1569867e+00, -3.5078707e+00, 3.7477229e+00],
[ 2.1569867e+00, 3.4406567e+00, 3.6862457e+00]],
[[-2.1319211e+00, 2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, -2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, 2.1319211e+00, -5.1160870e+00]],
[[ 0.0000000e+00, 3.1339090e+00, 3.1339090e+00],
[ 3.1339090e+00, 0.0000000e+00, 3.1339090e+00],
[ 3.1339090e+00, 3.1339090e+00, 0.0000000e+00]],
[[ 1.8203124e+00, -3.1528738e+00, 0.0000000e+00],
[ 1.8203124e+00, 3.1528738e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.0473022e+00]],
[[ 0.0000000e+00, 2.5252669e+00, 2.5252669e+00],
[ 2.5252669e+00, 0.0000000e+00, 2.5252669e+00],
[ 2.5252669e+00, 2.5252669e+00, 0.0000000e+00]],
[[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00]]], dtype=float32):Rendering array...
,), target_data=e_form=Array([-6.45586 , -6.8128195, -6.207207 , -5.4671555, -6.7837076,
-5.1435127, -7.477968 , -8.923818 , -8.200472 , -7.9467635,
-9.577978 , -6.3009315, -6.341573 , -6.54503 , -9.78936 ,
-8.184659 , -6.8508086, -5.8719406, -4.560977 , -6.476559 ,
-8.410168 , -7.7028112, -4.6160893, -7.076347 , -4.2489 ,
-4.8315697, -4.6079087, -5.067987 , -4.573908 , -4.5372405,
-3.712647 , 0. ], dtype=float32):Rendering array...
, force=Array([[ 0.01931927, -0.00101004, 0.00514533],
[ 0.01053436, 0.01622596, 0.00514533],
[ 0.00878491, -0.017236 , 0.00514533],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, stress=Array([[[-5.7650656e-01, 1.3470000e-05, 0.0000000e+00],
[-1.3470000e-05, -5.7650650e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8034226e-01]],
[[ 1.2198363e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0798175e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 7.2570233e+00]],
[[-6.6261089e-01, -1.3693068e+00, 1.7417374e-01],
[-1.3692706e+00, 9.3778890e-01, 5.2438201e-03],
[ 1.7419136e-01, 5.2425899e-03, -9.8206145e-01]],
[[-2.8648193e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -2.8648193e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8271584e+01]],
[[-3.5376236e+00, -6.9719963e-02, -1.5628552e-01],
[-6.9719352e-02, -3.4879370e+00, -6.0669428e-01],
[-1.5628555e-01, -6.0669398e-01, -2.6510825e+00]],
[[ 4.2228210e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 3.6939880e-01, 2.1828409e-01],
[ 0.0000000e+00, 2.1827926e-01, 7.3054400e-03]],
[[ 5.8439370e-02, 2.6000001e-07, -0.0000000e+00],
[-2.6000001e-07, 5.8439370e-02, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.5984979e+00]],
[[-5.6706039e+01, 2.8917530e+00, -1.5447336e+01],
[ 2.8917530e+00, -5.5068783e+01, -2.0427404e+01],
[-1.5447323e+01, -2.0427385e+01, -2.5514915e+01]],
[[-5.2726287e-01, -5.5939171e-02, -1.3698542e-01],
[-5.5939171e-02, -4.8771665e-01, -9.6870460e-02],
[-1.3698542e-01, -9.6870460e-02, -6.8537790e-01]],
[[ 5.0960288e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.7661177e+01, 8.4236950e-01],
[ 0.0000000e+00, 8.4236366e-01, -8.4243935e-01]],
[[-3.1354368e+00, -2.1732990e-02, -1.6962990e-02],
[-2.1732951e-02, -3.8033817e+00, 3.7452817e+00],
[-1.6962999e-02, 3.7452817e+00, -1.9895161e+00]],
[[ 1.7607585e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0304039e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.6413042e-01]],
[[-2.0985847e+00, -1.6143000e-04, -4.7509999e-05],
[-1.6518999e-04, -1.3785554e+00, -1.3842000e-04],
[-5.1700001e-05, -1.3332001e-04, 8.3491117e-01]],
[[-9.3856964e+00, -3.4404862e-01, -2.0616198e+00],
[-3.4404868e-01, -7.5551205e+00, -7.3723370e-01],
[-2.0616195e+00, -7.3723364e-01, -1.1444527e+01]],
[[-1.9861711e+00, -2.8908322e+00, 7.8711361e-01],
[-2.8947139e+00, -4.8037186e+00, 2.2988582e-01],
[ 7.8879166e-01, 2.2843994e-01, 2.1315180e+01]],
[[-1.4272230e+00, -1.6913697e+00, -5.8565631e+00],
[-1.6913695e+00, -4.9773030e+00, -1.3547415e+00],
[-5.8565636e+00, -1.3547415e+00, 1.7153270e+00]],
[[-6.0113930e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -8.9604294e+01, 7.7699997e-06],
[ 0.0000000e+00, 7.7699997e-06, -6.6187057e+01]],
[[ 2.7452025e+00, -0.0000000e+00, 4.3098168e+00],
[-0.0000000e+00, 9.5339581e-02, 0.0000000e+00],
[ 4.3098168e+00, 0.0000000e+00, -4.1897740e+00]],
[[-1.9521265e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -6.4751107e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.5113559e-01]],
[[ 2.3531018e+02, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.1068205e+01, -1.0828968e+02],
[ 0.0000000e+00, -1.0828968e+02, 1.9093062e+02]],
[[-4.9796581e+00, -1.2185392e+00, -4.9542028e-01],
[-1.2196398e+00, -2.3808525e+00, -6.0947609e-01],
[-4.9716067e-01, -6.1348146e-01, -7.8811878e-01]],
[[ 8.0838509e+00, 1.0610949e+01, -6.8462425e-01],
[ 1.0610948e+01, 1.6280436e+01, -5.4885068e+00],
[-6.8462473e-01, -5.4885073e+00, -1.0486948e+01]],
[[ 1.4801001e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6347537e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.6347537e+00]],
[[ 4.2995720e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995720e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7877603e+00]],
[[-5.6527023e+00, -1.9787040e-02, 0.0000000e+00],
[-1.9787040e-02, -5.6755505e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.4912200e+00]],
[[-1.5253462e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.5253462e+00, -0.0000000e+00],
[-0.0000000e+00, -0.0000000e+00, 3.4477546e+00]],
[[-1.2089447e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6878867e+00, -2.0085368e+00],
[ 0.0000000e+00, -2.0085406e+00, 3.9861345e+00]],
[[ 1.3792354e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.3792354e-01, -0.0000000e+00],
[ 0.0000000e+00, -0.0000000e+00, 2.5862169e-01]],
[[ 1.4818479e+01, 1.4620157e+00, 1.4620157e+00],
[ 1.4620157e+00, 1.4818479e+01, 1.4620157e+00],
[ 1.4620157e+00, 1.4620157e+00, 1.4818479e+01]],
[[-3.0764341e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -3.0764344e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, 4.8425870e+00]],
[[-1.4238837e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.4238837e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, -1.4238837e+00]],
[[ 1.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0000000e+00]]], dtype=float32):Rendering array...
,),), 'energies': Array([-1.3227775e+00, -1.3328015e+00, -1.3621812e+00, ...,
-5.4650640e-44, -5.6051939e-44, -5.0446745e-44], dtype=float32):Rendering array...
,}children: 'reduction': reductioninput: 'data': Array([-8.599918, -8.603131, -8.612547, ..., 0. , 0. ,
0. ], dtype=float32):Rendering array...
, 'segments': Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
, 'num_segments': 32,}output: Array([-6.459608 , -6.81614 , -6.209894 , -5.464156 , -6.7733355,
-5.1489067, -7.4866276, -8.940734 , -8.208114 , -7.9659204,
-9.582554 , -6.2796736, -6.3338375, -6.549851 , -9.774819 ,
-8.180547 , -6.862594 , -5.863358 , -4.5622835, -6.4760146,
-8.4053755, -7.6991196, -4.6202903, -7.042828 , -4.251147 ,
-4.8484707, -4.60405 , -5.0757895, -4.5812078, -4.432643 ,
-3.6734219, 0. ], dtype=float32):Rendering array...
,),}params: 180 parametersscale: Array([4.2038954e-45, 3.0043125e-01, 4.1919053e-01, 3.2384700e-01,
3.2050502e-01, 3.0024481e-01, 2.6050037e-01, 4.1325513e-01,
2.9645354e-01, 2.7672523e-01, 7.3047429e-01, 2.7780551e-01,
4.0432665e-01, 2.6659670e-01, 2.9438594e-01, 4.3402749e-01,
3.4227425e-01, 2.9799023e-01, 7.4895459e-01, 2.5004971e-01,
3.5872504e-01, 2.8236023e-01, 6.4121890e-01, 3.9409107e-01,
7.4890226e-01, 3.5291594e-01, 3.1090504e-01, 3.4960249e-01,
4.6771184e-01, 5.9428304e-01, 2.5766170e-01, 2.8141865e-01,
4.9121195e-01, 3.2623026e-01, 6.4953142e-01, 6.3061160e-01,
8.5306853e-01, 6.3829052e-01, 7.6778311e-01, 2.8485122e-01,
2.8993279e-01, 2.7547282e-01, 9.6464241e-01, 7.5048459e-01,
8.1315958e-01, 7.9247254e-01, 4.5107767e-01, 4.5357430e-01,
4.8774320e-01, 3.5615650e-01, 2.5400338e-01, 3.0025351e-01,
3.7530035e-01, 6.5497804e-01, 2.7789482e-01, 3.1669718e-01,
4.0909296e-01, 3.8105384e-01, 7.5613052e-01, 7.5295144e-01,
8.0258620e-01, 7.9740274e-01, 1.5784103e-01, 6.9885850e-01,
7.6217997e-01, 5.1847076e-01, 3.0885923e-01, 5.0938869e-01,
5.6917423e-01, 7.5945461e-01, 6.3645500e-01, 5.9862715e-01,
5.5500871e-01, 7.4034333e-01, 3.0939424e-01, 5.0928408e-01,
7.7029741e-01, 2.4929152e-01, 7.7238590e-01, 7.6192188e-01,
6.0438299e-01, 1.8188974e+00, 1.0024991e+00, 7.7957290e-01,
7.7421695e-01, 1.9463260e+00, 8.6763585e-01, 6.9731098e-01,
8.5510230e-01, 4.2600378e-01], dtype=float32):Rendering array...
, shift: Array([ 0. , -8.254613 , -3.8793902 , -2.9042175 ,
-8.1759615 , -6.3587766 , -4.0522203 , -1.9677219 ,
-7.945281 , -3.5150082 , -8.059753 , -3.1420217 ,
-1.8107116 , -3.5583389 , -8.935444 , -9.42458 ,
-9.403597 , -5.4425173 , -6.7757645 , -4.5880127 ,
-5.1307225 , -4.574444 , -3.315387 , -1.9589221 ,
-6.82099 , -5.8368945 , -5.44436 , -8.114189 ,
-6.29539 , -3.7478168 , -1.1364459 , -3.721388 ,
-10.180324 , -6.538427 , -3.8525372 , -3.5311642 ,
-11.642563 , -10.856619 , -8.515128 , -2.2322743 ,
-0.6706738 , -3.884738 , -9.75878 , -10.200178 ,
-12.651368 , -7.912676 , -7.1350646 , -1.7933267 ,
-4.5861883 , -2.4339597 , -4.706932 , -3.4705129 ,
-7.2555256 , -9.282878 , -2.6069057 , -2.035008 ,
-3.1942008 , -6.1275 , -6.5005355 , -12.746008 ,
-7.2460003 , -13.921683 , -3.484169 , -15.289538 ,
-6.6684675 , -8.383291 , -2.104009 , -4.239751 ,
-11.927877 , -16.182364 , -12.089612 , -11.375482 ,
-9.446338 , -6.8297224 , -2.9690852 , -8.491434 ,
-6.6075172 , 0.03989757, -6.690451 , -6.6346316 ,
-9.987788 , 5.55694 , -11.5706415 , -6.5398293 ,
-6.751645 , 6.2271156 , -6.290777 , -0.219617 ,
-1.8358347 , -0.51835036], dtype=float32):Rendering array...
,}output: Array([[-6.459608 ],
[-6.81614 ],
[-6.209894 ],
[-5.464156 ],
[-6.7733355],
[-5.1489067],
[-7.4866276],
[-8.940734 ],
[-8.208114 ],
[-7.9659204],
[-9.582554 ],
[-6.2796736],
[-6.3338375],
[-6.549851 ],
[-9.774819 ],
[-8.180547 ],
[-6.862594 ],
[-5.863358 ],
[-4.5622835],
[-6.4760146],
[-8.4053755],
[-7.6991196],
[-4.6202903],
[-7.042828 ],
[-4.251147 ],
[-4.8484707],
[-4.60405 ],
[-5.0757895],
[-4.5812078],
[-4.432643 ],
[-3.6734219],
[ 0. ]], dtype=float32):Rendering array...
,),}output: Array([[-6.459608 ],
[-6.81614 ],
[-6.209894 ],
[-5.464156 ],
[-6.7733355],
[-5.1489067],
[-7.4866276],
[-8.940734 ],
[-8.208114 ],
[-7.9659204],
[-9.582554 ],
[-6.2796736],
[-6.3338375],
[-6.549851 ],
[-9.774819 ],
[-8.180547 ],
[-6.862594 ],
[-5.863358 ],
[-4.5622835],
[-6.4760146],
[-8.4053755],
[-7.6991196],
[-4.6202903],
[-7.042828 ],
[-4.251147 ],
[-4.8484707],
[-4.60405 ],
[-5.0757895],
[-4.5812078],
[-4.432643 ],
[-3.6734219],
[ 0. ]], dtype=float32):Rendering array...
,)